poj 2299 Ultra-QuickSort (树状数组求逆序数）

/*********************************************
题目：    Ultra-QuickSort（poj 2299）
链接：    http://poj.org/problem?id=2299
题意：    求逆序对。
算法：    树状数组+离散化。
***********************************************/
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;

const int mx=500005;
struct T
{
    int x,di;
};
T s[mx];
int cut;
int c[mx];
long long p[mx];

bool com1(T a,T b)
{
    return a.x<b.x;
}
bool com2(T a,T b)
{
    return a.di<b.di;
}

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

void updata(int x)
{
    if (x>cut) return ;
    c[x]++;
    updata(x+lowbit(x));
}

int sum(int x)
{
    if (x<=0) return 0;
    return c[x]+sum(x-lowbit(x));
}

int main()
{
    int n;
    while (scanf("%d",&n)&&n)
    {
        for (int i=0;i<n;i++)
        {
            scanf("%d",&s[i].x);
            s[i].di=i;
        }
        sort(s,s+n,com1);
        int temp=s[0].x;
        s[0].x=cut=1;
        for (int i=1;i<n;i++)
        {
            if (s[i].x==temp) s[i].x=cut;
            else
            {
                temp=s[i].x;
                s[i].x=++cut;
            }
        }
        sort(s,s+n,com2);
        memset(c,0,sizeof(c));
        long long ans=0;
        for (int i=0;i<n;i++)
        {
            ans+=i-sum(s[i].x);
            updata(s[i].x);
        }
        printf("%lld
",ans);
    }
}