POJ 2299 Ultra-QuickSort（树状数组）

题目链接：http://poj.org/problem?id=2299

求逆序数，可以用归并排序求，这里用树状数组。因为数字最大可以达到999999999，开这么大的数组肯定是不现实的。所以要先把数据离散化处理。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
int c[500005],a[500005],n;
struct node
{
    int val,pos;
}b[500005];

bool cmp(node a,node b)
{
    return a.val<b.val;
}
int lowbit(int x)
{
    return x&-x;
}

void update(int p,int v)
{
    while(p<=n)
    {
        //printf("111\n");
        c[p]+=v;
        p+=lowbit(p);
    }
}

int sum(int p)
{
    int ans=0;
    while(p>0)
    {
        ans+=c[p];
        p-=lowbit(p);
    }
    return ans;
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        ll ans=0;
        memset(c,0,sizeof(c));
        if(n==0) break;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&b[i].val);
            b[i].pos=i;
        }
        sort(b+1,b+n+1,cmp);
        for(int i=1;i<=n;i++)
            a[b[i].pos]=i;
        for(int i=1;i<=n;i++)
        {
            update(a[i],1);
         //   printf("sum=%d\n",sum(a[i]));
            ans+=i-sum(a[i]);
        }
        printf("%lld\n",ans);
    }
}