poj 2299 Ultra-QuickSort(离散化+树状数组)

按照 http://hi.baidu.com/czyuan_acm/blog/item/49f02acb487f06f452664fbc.html的思路写的代码,

1.先离散化,要倒着编号

2.求出每个点左边比它小的点的个数


代码:

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

struct node{
    int value,no;
}arr[1000100];
int tree[1000100];
int b[1000100];
int n;
int cmp(node x,node y)
{
	return x.value<y.value;
}
int LowBit(int x)//当x=0时下面的函数死循环,LowBit(0)=0;
{
	return x&(-x);
}
int GetSum(int x)
{
	int temp=0;
   for(int i=x;i>=1;i-=LowBit(i))
	   temp+=tree[i];
   return temp;
}
void UpDate(int x,int c)
{
	for(int i=x;i<=n;i+=LowBit(i))
		tree[i]+=c;
}
int main()
{
	while(scanf("%d",&n)!=EOF && n!=0)
	{
		memset(tree,0,sizeof(tree));
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&arr[i].value);
			arr[i].no=i;
		}
		sort(arr+1,arr+1+n,cmp);
		for(int i=1;i<=n;i++)
		{
			arr[arr[i].no ].value=n-i+1;//倒着编号,离散化
		}
		for(int i=1;i<=n;i++)
		{
			b[i]=GetSum(arr[i].value);//查找每个点左边比它小的 点个数
			UpDate(arr[i].value,1);//插入一个点
		}
		__int64 ans=0;
		for(int i=1;i<=n;i++)
			ans+=b[i];
		printf("%I64d\n",ans);
	}
	system("pause");
	return 0;
}


你可能感兴趣的:(poj 2299 Ultra-QuickSort(离散化+树状数组))