POJ 2299 -Ultra-QuickSort-树状数组求逆序数

POJ 2299 Ultra-QuickSort

使用树状数组记录逆序对数。

把数组按照大小顺序插入，getsum(i)就是i前面的比他大的数。

#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn = 500005;
int reflect[maxn],c[maxn],N;

struct Node
{
    int val;
    int pos;
    bool operator < (const Node &r) const
    {
        return val < r.val;
    }
}node[maxn];

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

void update(int x)
{
    while(x<=N)
    {
        c[x] += 1;
        x += lowbit(x);
    }
}

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

int main()
{
    while(~scanf("%d",&N) && N)
    {
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&node[i].val);
            node[i].pos = i;
        }
        sort(node+1,node + N+1);
        for(int i=1;i<=N;i++) reflect[node[i].pos] = i;
        memset(c,0,sizeof c);

        long long ans = 0;
        for(int i=1;i<=N;i++)
        {
            update(reflect[i]);
            ans += i - getsum(reflect[i]);
        }
        printf("%lld
",ans);
    }
}