首页 > 编程语言 > 详细

[数据结构-树状数组]POJ 2299 Ultra-QuickSort

时间:2016-03-16 18:40:57      阅读:239      评论:0      收藏:0      [点我收藏+]

题目大意就是要求数组中逆序数的对数,和51nod1019一样,可以用树状数组。

先对所有数进行排序,还要记录每个数原来在数组中的下标,然后对每个数data[i]计算出值小于data[i]且原始位置下标大于data[i]原始下标的数个数。

在新数组中,从小到大,动态更新数组C。假设更新到了data[i],C[k]表示的是在当前已经访问过的数中(即小于data[i]的所有数)中原始下标小于等于k的个数。

同时动态查询值小于data[i]且原始位置下标大于data[i]原始下标的数个数。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int C[500005],n;
int lowbit(int x)
{
    return x&-x;
}
int sum(int x)      //求前缀和,x向左上爬
{
    int ret=0;
    while(x>0)
    {
        ret+=C[x];
        x-=lowbit(x);
    }
    return ret;
}
void add(int x,int d)     //A[x]加上d,x向右上爬
{
    while(x<=n)
    {
        C[x]+=d;
        x+=lowbit(x);
    }
}
struct D
{
    int id;
    long long val;
};
bool operator < (D a,D b)
{
    return a.val<b.val;
}
D data[500005];
int main()
{
    while(cin>>n&&n)
    {
        long long ans=0;
        for(int i=1; i<=n; i++)
        {
            cin>>data[i].val;
            data[i].id=i;
        }
        sort(data+1,data+n+1);
        data[0].val=-111;
        memset(C,0,sizeof(C));
        int pre;
        for(int i=1; i<=n; i++)
        {
            if(data[i].val!=data[i-1].val)
            {
                pre=sum(n)-sum(data[i].id);
                ans+=pre;
            }
            else ans+=pre;
            add(data[i].id,1);
        }
        cout<<ans<<endl;
    }
}

[数据结构-树状数组]POJ 2299 Ultra-QuickSort

原文:http://www.cnblogs.com/lastone/p/5284491.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!