洛谷P1908逆序对

题目链接
洛谷P1908逆序对
题目大意:
对于序列 a [ N ] a[N] ,求满足 i < j i<j a [ i ] > a [ j ] a[i]>a[j] 的逆序对总数
输入输出样例
输入

6
5 4 2 6 3 1

输出

11

思路:
利用归并排序求逆序对。
对于上述例子,当部分排好序之后可为2,4,5,1,3,6,以此时为例,i=1,j=4,mid=3,此时由于a[i]>a[j]&&i<j,此时a[]从i到mid部分均比a[j]大,逆序对总数ans+=mid-i+1。以此类推。

#include<iostream>
#include<cstdio>
#define N 500010
long long n,ans=0;
int a[N],c[N];
void msort(int l,int r){
	if(l==r) return;
	int mid=(l+r)>>1;
	msort(l,mid);msort(mid+1,r);
//	int i=l,j=mid+1,k=1;
	int i=l,j=mid+1,k=l;//确保前面部分已经排好序 
	while(i<=mid&&j<=r){
		if(a[i]<=a[j]){
			c[k++]=a[i++];
		}
		else{
			c[k++]=a[j++];
			ans+=mid-i+1;
		}
	}
	while(i<=mid)
		c[k++]=a[i++];
	while(j<=r)
		c[k++]=a[j++];
		
	for(int d=l;d<=r;d++)//排好序 
		a[d]=c[d];
}
int main(int argc, char** argv) {
	int cnt=0;
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	msort(1,n);
	printf("%lld\n",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/zhuixun_/article/details/82801079