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

Ultra-QuickSort

 题目链接:POJ - 2299

题意:给出一个数列,通过不断交换相邻元素使得数列有序,问最少需要交换几次;实际上就是逆序对的个数;

用树状数组求逆序对,此题需要先将数列离散化一下,因为原数列不是连续的,而且范围太大;

倒序插入,每次求插入该数之前前边已经有几个比他小的数已经存在了;

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int maxn=500010;
struct node{
	int val, x;
	bool operator < (const node &p) const {
		return val<p.val;
	}
}input[maxn];
int tr[maxn<<2];
int op[maxn];
int lowbit(int x){
	return x&(-x);
}
int n;
void updata(int x){
	while(x<(n<<1)){
		tr[x]++;
		x+=lowbit(x);
	}
}
long long getSum(int x){
	long long s=0;
	while(x>0){
		s+=tr[x];
		x-=lowbit(x);
	}
	return s;
}
int main(){
	while(scanf("%d", &n), n){
		for(int i=1; i<=n; i++){
			scanf("%d", &input[i].val);
			input[i].x=i;
		}
		sort(input+1, input+1+n);
		for(int i=1; i<=n; i++){
			op[input[i].x]=i;
		}
		long long ans=0;
		memset(tr, 0, sizeof(tr));
		for(int i=n; i>0; i--){
			updata(op[i]);
			ans+=getSum(op[i]-1);
		}
		printf("%lld\n", ans);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/sirius_han/article/details/81121296