POJ 2299 Ultra-QuickSort【树状数组+离散化】

题意: 给你一组数,让你求他的逆序数。

分析: 数子比较大,直接用数组存不下,所以要进行离散话坐标。

View Code
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct node
{
    int xu;
    int num;
}q[500001];
int cmp(const void*p1,const void*p2)
{
    struct node*c=(struct node*)p1;
    struct node*d=(struct node*)p2;
    return c->num-d->num;
}
int n;
int a[500001];
int b[500001];
int lowbit(int x)
{
    return (x)&(-x);
}
void add(int pos)
{
    while(pos<=n)
    {
        a[pos]+=1;
        pos+=lowbit(pos);
    }
}
int getsum(int pos)
{
    int sum=0;
    while(pos>0)
    {
        sum+=a[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
int main()
{
    int i,tot;
    long long res;
    while(scanf("%d",&n),n)
    {
        for(i=0;i<n;i++)
        {
            scanf("%d",&q[i].num);
            q[i].xu=i;
        }
        qsort(q,n,sizeof(q[0]),cmp);
        tot=1;
        res=0;
        b[q[0].xu]=tot;
        for(i=1;i<n;i++)
        {
            if(q[i].num==q[i-1].num)
                b[q[i].xu]=tot;
            else
                b[q[i].xu]=++tot;
        }
        for(i=0;i<=tot;i++)
            a[i]=0;
        for(i=0;i<n;i++)
        {
            res+=(long long)getsum(tot)-(long long)getsum(b[i]);
            add(b[i]);
        }
        printf("%lld\n",res);
    }
    return 0;
}
posted @ 2012-05-03 21:15  'wind  阅读(181)  评论(0)    收藏  举报