1203: 逆序数【归并排序】

1203: 逆序数

Description

在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数不小于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。

如2 4 3 1中，2 1，4 3，4 1，3 1是逆序，逆序数是4。给出一个整数序列，求该序列的逆序数。

Input

多组测试数据

每组测试数据分两行，第一行一个正整数n（n <= 50000)

第二行有n个元素( 0 <= A[i] <= 10^9)

Output

每组测试数据输出一行表示逆序数

Sample Input

4 2 4 3 1 3 1 1 1

Sample Output

4 3

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int a[maxn],temp[maxn],cnt;
void mergearray(int l,int mid,int r)
{
	//cout<<cnt<<endl;
	int i=l,j=mid+1,m=mid,n=r,k=0;
	while(i<=m&&j<=n)
	{
		if(a[i]<a[j])
		temp[k++]=a[i++];
		else 
		{
			cnt+=m-i+1;
			temp[k++]=a[j++];
		}
	}
	while(i<=m) temp[k++]=a[i++];
	while(j<=n) temp[k++]=a[j++];
	for(i=l;i<=r;i++)
		a[i]=temp[i-l];
}
void mergeSort(int l,int r)
{
	int mid=(l+r)>>1;
	//	cout<<l<<" "<<mid<<" "<<r<<endl; 
	if(l<r)
	{
		mergeSort(l,mid);
		mergeSort(mid+1,r);
		mergearray(l,mid,r);
	//	cout<<l<<","<<mid<<","<<r<<endl;
	}
}
int main(void)
{
	int n;
	while(~scanf("%d",&n))
	{
		cnt=0;
		memset(a,0,sizeof a);
		memset(temp,0,sizeof temp);
		for(int i=0;i<n;i++)
			scanf("%d",&a[i]);
		mergeSort(0,n-1);
		printf("%d\n",cnt);
	}
	return 0;
}