关闭

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

标签: 树状数组归并
221人阅读 评论(0) 收藏 举报
分类:

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

题意:求逆序数。

分析:以前用归并写过,也可以用树状数组求,不过要离散化一下。简单说一下,对于每一个数b[i],求出它后面有多少个数比它小,就是查询[1,b[i]-1](离散化后)有多少个数存在就行了。更新,更新所有管辖b[i]的管辖点。

代码:

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

const int maxn = 5e5+5;
int tree[maxn],a[maxn],b[maxn],n,tail;

int Find(int x)
{
	int down=1,mid,up=tail;
	while(down<=up)
	{
		mid=(down+up)>>1;
		if(a[mid]>x)
			up=mid-1;
		else if(a[mid]==x)
			return mid;
		else
			down=mid+1;
	}
	return mid;
}

int lowbit(int x)
{
	return x&-x;
}

void update(int x,int v)
{
	for(int i=x;i<=n;i+=lowbit(i))
		tree[i]+=v;
}

int query(int x)
{
	int ret(0);
	for(int i=x;i>0;i-=lowbit(i))
		ret+=tree[i];
	return ret;
}

int main() 
{
	int i,temp;
	while(scanf("%d",&n),n)
	{
		memset(tree,0,sizeof(tree[0])*(n+2));
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			b[i]=a[i];
		}
		sort(a+1,a+n+1);
		tail=unique(a+1,a+n+1)-(a+1);
		long long ans=0;
		for(i=n;i>=1;i--)
		{
			temp=Find(b[i]);
			update(temp,1);
			ans+=query(temp-1);
		}
		printf("%lld\n",ans);
	}
	return 0;
}

归并代码:

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

const int maxn = 5e5+5;
int a[maxn],ta[maxn];
long long ans=0;

void merge_sort(int A[],int x,int y,int t[])
{
	if(y>x)
	{
		int m=(x+y)>>1;
		merge_sort(A,x,m,t);
		merge_sort(A,m+1,y,t);
		int p=x,q=m+1,i=x;
		while(p<=m || q<=y)
		{
			if(q>y || p<=m && A[p]<=A[q])
				t[i++]=A[p++];
			else
			{
				t[i++]=A[q++];
				ans+=q-i;
			}
		}
		for(i=x;i<=y;i++)
			A[i]=t[i];
	}
}

int main()
{
	int n;
	while(scanf("%d",&n),ans=0,n)
	{
		for(int i=0;i<n;i++)
			scanf("%d",&a[i]);
		merge_sort(a,0,n-1,ta);
		printf("%lld\n",ans);
	}
	return 0;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:69323次
    • 积分:2641
    • 等级:
    • 排名:第13815名
    • 原创:201篇
    • 转载:2篇
    • 译文:0篇
    • 评论:5条
    最新评论