POJ3067 Japan 树状数组

Japan某城市的左边从上到下有n个城市，右边从上到下有m个城市，

现要建k条高铁，起点在左边，终点在右边，

问有多少个交点（任一交点只能有2条线同时经过）。

排序+树状数组，插点问段。

排序后，按起点从小到大扫一遍，

a[j] 表示前i条线中，终点为j的个数。

c[j] a[j]的树状数组。

注意：最后结果要用long long 

#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=1000+5;
int c[MAXN];
struct Edge
{
    int u,v;
}edge[MAXN*MAXN];
int n,m;
bool cmp(Edge i,Edge j)
{
    if(i.u==j.u)
        return i.v<j.v;
    return i.u<j.u;
}
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int temp=0;
    while(x>0){
        temp+=c[x];
        x-=lowbit(x);
    }
    return temp;
}
void update(int x,int num)
{
    while(x<=m){
        c[x]+=num;
        x+=lowbit(x);
    }
}
int main()
{
    int test;
    int cas=0;
    scanf("%d",&test);
    while(test--){
            cas++;
        int k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=1;i<=k;i++)
            scanf("%d%d",&edge[i].u,&edge[i].v);
        sort(edge+1,edge+k+1,cmp);
        memset(c,0,sizeof(c));
        long long  ans=0;
        for(int i=1;i<=k;i++){
            update(edge[i].v,1);
            ans+=(long long)(i-sum(edge[i].v));
        }
        printf("Test case %d: %I64d
",cas,ans);
    }
    return 0;
}