POJ 3067 Japan(树状数组)

题意:左边有N个点,右边有M个点,有K条线连接左边的点和右边的点,问所有线的交点一共有多少个。

题记:树状数组求逆序对。

#include
#include
#include
#include
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
typedef long long ll;
const int N=1000;
ll s[N+10];
struct Node{
    int l,r;
    bool operator < (const Node &W)const{
        if(l==W.l) return r>W.r;
        else return l>W.l;
    }
}a[N*N+10];

int lowbit(int x){
    return x&(-x);
}

void update(int x,int d){
    while(x<=N){
        s[x]+=d;
        x+=lowbit(x);
    }
}

ll getsum(int x){
    ll sum=0;
    while(x>0){
        sum+=s[x];
        x-=lowbit(x);
    }
    return sum;
}

int main(){
    int T,cas=1;
    scanf("%d",&T);
    while(T--){
        int n,m,k;
        scanf("%d%d%d",&m,&n,&k);
        memset(s,0,sizeof(s));
        for(int i=1;i<=k;i++)
            scanf("%d%d",&a[i].l,&a[i].r);
        sort(a+1,a+1+k);
        ll ans=0;
        for(int i=1;i<=k;i++){
            ans+=getsum(a[i].r-1);
            update(a[i].r,1);
        }
        printf("Test case %d: %lld\n",cas++,ans);
    }
    return 0;
}

你可能感兴趣的:(POJ)