ACM PKU 2299 Ultra-QuickSort

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

这道题是求逆序数的,离散化+树状数组;

注意的地方:1、离散化的比较函数;2、结果会爆int,所以必须用__int64或者是long long 类型来存储结果;

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

const int MAXN=500000;
int n_case;
int B[MAXN+10],C[MAXN+10];
struct node
{
    int index,val;
} A[MAXN+10];

int cmp(const void * a,const void * b)     //离散化比较函数模板;这里解决了一些相同的情况
{
        return ((node *)a)->val - ((node *)b) ->val;
}

int lowbit(int a)    //树状数组的关键地方;
{
    return (-a)&a;
}

void insert(int x)
{
    for(int i=x; i<MAXN; i+=lowbit(i))
    {
        C[i]++;
    }
}

int getSum(int x)
{
    int ans=0;
    for(int i=x; i>0; i-=lowbit(i))   ans+=C[i];
    return ans;
}

int main()
{
    //freopen("in.txt","r",stdin);
    long long sum;           //用long long存储计算结果;
    while(scanf("%d",&n_case)&&n_case!=0)
    {
        int temp;
        sum=0;
        for(int i=1; i<=n_case; i++)
        {
            scanf("%d",&temp);
            A[i].val=temp;
            A[i].index=i;
        }
        qsort(A+1,n_case,sizeof(A[0]),cmp);

        for(int i=1; i<=n_case; i++)
        {
            B[A[i].index]=i;
        }
        memset(C,0,sizeof(C));
        for(int i=n_case; i>0; i--)   //这里从最后一个开始,查找放进树状数组C里比当前元素小的元素,即 前面所有之和;因为树状数组存储的是逆序数;
        {
            sum+=getSum(B[i]);
            insert(B[i]);
        }
        printf("%I64d\n",sum);
    }
    return 0;
}

  

posted on 2011-08-20 10:20  _Clarence  阅读(132)  评论(0编辑  收藏  举报

导航