首页 > 其他 > 详细

PKU Ultra-QuickSort (树状数组求逆序数)

时间:2014-04-26 12:29:11      阅读:530      评论:0      收藏:0      [点我收藏+]


题意:就是求逆序数;


思路:我们用一个结构体保存数列的值和下标(num,id);

然后对 num 从大到小排列,算出num之前比num的id还要小的个数,即a[i] > a[j] && i < j 为一对逆序数;



#include<cstdio>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<map>
#include<cmath>
#include<iostream>
#include <queue>
#include <stack>
#include<algorithm>
#include<set>
using namespace std;
#define INF 1e8
#define eps 1e-8
#define ll __int64
#define maxn 500010
#define mol 1000000007

struct node
{
	int id,num;
}a[maxn];
int c[maxn],b[maxn];
int lowbit(int n)
{
	return n&(-n);
}
void update(int x,int k)
{
	while(x<=maxn)
	{
		c[x]+=k;
		x+=lowbit(x);
	}
}
ll getsum(int x)
{
	ll sum=0;
	while(x>0)
	{
		sum+=c[x];
		x-=lowbit(x);
	}
	return sum;
}
int cmp(node a,node b)
{
	return a.num>b.num;
}
int main()
{
	int n;
	while(scanf("%d",&n)&&n)
	{
		ll ans=0;
		memset(c,0,sizeof(c));
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].num);
			a[i].id=i;

		}
		sort(a+1,a+n+1,cmp);
		for(int i=1;i<=n;i++)
		{
			b[a[i].id]=i;
		}
		for(int i=1;i<=n;i++)
		{
			update(b[i],1);
			//printf("%d\n",getsum(b[i]));
			ans+=(getsum(b[i])-1);//减去自己的数
		}
		printf("%I64d\n",ans);
	}
	return 0;
}


PKU Ultra-QuickSort (树状数组求逆序数),布布扣,bubuko.com

PKU Ultra-QuickSort (树状数组求逆序数)

原文:http://blog.csdn.net/u012861385/article/details/24519243

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