题目(洛谷P1908)
最近,TOM 老猫查阅到一个人类称之为“逆序对”的东西,这东西是这样定义的:对于给定的一段正整数序列,逆序对就是序列中 a_i>a_jai>aj 且 i<ji<j 的有序对。知道这概念后,他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。
思想
利用分治思想的归并排序来解决这道问题。
若一个数组从中间(mid)若分为两个升序的部分,当左边数组的某一个值(下标为i)大于右边数组的某一个值(下标为j)时,则左边数组下标大于i的值都会比下标为j的值大,即会产生(mid-i+1)个逆序对。例如2,4,5和1,3,6,因为2大于1,所以2之后的数字(4和5)也一定大于1,于是产生了mid-i+1(3-1+1=3)个逆序对。
当数组分为两部分时,其中一个部分中的数字位置进行了交换并不会影响另外一部分与该部分之间产生的逆序对数,因此利用归并排序将一个数组分到最小(一个数字就是一个部分),再进行归并,就可以找到数组中所有部分之间产生的总的逆序对数。
方法
将数组[1,2,3...n]分为[1,...mid]和[mid+1,n],我们直接假设这两部分是有序(从小到大)的,分解的步骤留给递归去做就可以了。还需要开辟一个等大的数组保存归并后的顺序,代码如下:
#include<iostream>
using namespace std;
int n,a[500010],c[500010];
long long ans;
void msort(int b,int e){
int i = b,k = b,mid = (b+e) / 2,j = mid + 1;
if(b == e) return ;
msort(b,mid);
msort(mid+1,e);//递归分解数组
while(i<=mid&&j<=e){
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<=e) c[k++] = a[j++];
for(int x = b;x<=e;x++){
a[x] = c[x];
}
}
int main(){
cin>>n;
for(int i = 1;i<=n;i++){
cin>>a[i];
}
msort(1,n);
cout<<ans;
}