二分·归并排序之逆序对
题目链接:
http://hihocoder.com/problemset/problem/1141
官方题解:
我们来看一个归并排序的过程:
给定的数组为[2, 4, 5, 3, 1],二分后的数组分别为[2, 4, 5], [1, 3],假设我们已经完成了子过程,现在进行到该数组的“并”操作:
a: [2, 4, 5] | b: [1, 3] | result:[1] | 选取b数组的1 | |||
a: [2, 4, 5] | b: [3] | result:[1, 2] | 选取a数组的2 | |||
a: [4, 5] | b: [3] | result:[1, 2, 3] | 选取b数组的3 | |||
a: [4, 5] | b: [] | result:[1, 2, 3, 4] | 选取a数组的4 | |||
a: [5] | b: [] | result:[1, 2, 3, 4, 5] | 选取a数组的5 |
在执行[2, 4, 5]和[1, 3]合并的时候我们可以发现,当我们将a数组的元素k放入result数组时,result中存在的b数组的元素一定比k小。
在原数组中,b数组中的元素位置一定在k之后,也就是说k和这些元素均构成了逆序对。
那么在放入a数组中的元素时,我们通过计算result中b数组的元素个数,就可以计算出对于k来说,b数组中满足逆序对的个数。
又因为递归的过程中,a数组中和k满足逆序对的数也计算过。则在该次递归结束时,[2, 4, 5, 3, 1]中所有k的逆序对个数也就都统计了。
同理对于a中其他的元素也同样有这样的性质。
由于每一次的归并过程都有着同样的情况,则我们可以很容易推断出:
若将每一次合并过程中得到的逆序对个数都加起来,即可得到原数组中所有逆序对的总数。
即在一次归并排序中计算出了所有逆序对的个数,时间复杂度为O(NlogN)。
AC代码(归并排序):
#include <bits/stdc++.h>
#define INF 0x7fffffff
//我去,一开始写成0xfffffff,一直不知道那错了。。。
using namespace std;
typedef long long ll;
const int N = 100005;
int a[N];
int L[N],R[N];
ll cnt;
void Merge(int a[],int l,int mid,int r){
int n1 = mid-l+1;
int n2 = r-mid;
int i,j,k;
for(i = 1; i <= n1; i++)
L[i] = a[l+i-1];
for(i = 1; i <= n2 ;i++)
R[i] = a[mid+i];
L[n1+1] = INF;
R[n2+1] = INF;
i = 1;
j = 1;
for(k = l; k <= r; k++){
if(L[i] <= R[j]){
a[k] = L[i];
i++;
}
else{
a[k] = R[j];
j++;
cnt += (ll)(n1-i+1);
}
}
}
void Merge_sort(int a[],int l,int r){
if(l < r){
int mid = (l+r)>>1;
Merge_sort(a,l,mid);
Merge_sort(a,mid+1,r);
Merge(a,l,mid,r);
}
}
int main(){
int n;
while(~scanf("%d",&n)){
cnt = 0;
for(int i = 1; i <= n; i++)
scanf("%d",&a[i]);
Merge_sort(a,1,n);
printf("%lld\n",cnt);
}
return 0;
}