树状数组_poj_2299_Ultra-QuickSort

题目链接

题意：求逆序对。
方法：树状数组+离散化

这道题数据量比较大，直接开数组是不行的，所以采取离散化的办法，其实也就是弄一个映射关系而已。

具体思想就是排好序，建立映射关系，然后开一个一样大的数组，按照原来的输入顺序在其映射的位置加1 （加的时候用树状数组的更新函数）,然后用区间查询函数统计它前面有多少个数已经输入了（即区间求和），在这个数前面的都是比它小的数，所以算一下就知道有多少比它大的数在它前面了，即逆序对求出。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm> 
using namespace std;
const int maxn=500000+50;

int a[maxn],ref[maxn];
int N;
struct node{int v,pos;}nod[maxn];
bool cmp(node x,node y){
    return x.v<y.v;
}

void Add(int k){
    while(k<=N){
        a[k]+=1;
        k+=k&-k;
    }
}

int query(int x){
    int ans=0;
    while(x){
        ans+=a[x];
        x-=x&-x;
    }
    return ans;
}

int main(){
    while(cin>>N&&N){
        for(int i=1;i<=N;i++){
            scanf("%d",&nod[i].v);
            nod[i].pos=i;
        }
        memset(a,0,sizeof(a));
        sort(nod+1,nod+1+N,cmp);
        for(int i=1;i<=N;i++)ref[nod[i].pos]=i;
        long long ans=0;
        for(int i=1;i<=N;i++){
            Add(ref[i]);
            ans+=i-query(ref[i]);
        }
        cout<<ans<<endl;
    }
    return 0;
}