上一篇是用类似于“桶排序”的方法,利用树状数组求和,得出的逆序数 。
用归并排序来求,也是相似的道理,都是求在前面比自己大的数有几个。归并排序是稳定排序,所以可以这么做(插入排序和希尔排序按理也可以,只是效率更没那么高)。归并排序在归并两个子段的时候,子段都是已经排好序的(因为回溯),相对位置在右边的子段的元素如果小于相对位置在左边的子段的元素,说明在原序列中,左边更大的元素肯定也在它的前面,从这个左边更大元素开始往右的部分都是更大的,而且原序列中也一定在它的前面,这些都可以构成逆序数。
核心代码是:
while( i <= mid && j <= r ){
if( a[i] <= a[j] )
help[k++] = a[i++] ;
else
help[k++] = a[j++] , ans += mid-i+1 ;
}
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std ;
typedef long long LL ;
LL ans ;
int n , a[500005] ;
int help[500005] ;
void Merge( int l , int mid , int r ){
int i = l , j = mid+1 , k = 0 ;
while( i <= mid && j <= r ){
if( a[i] <= a[j] )
help[k++] = a[i++] ;
else
help[k++] = a[j++] , ans += mid-i+1 ; // 左边的大于右边的,在原序列中也是一样,构成逆序数
}
while( i <= mid )
help[k++] = a[i++] ;
while( j <= r )
help[k++] = a[j++] ;
for( int t = 0 ; t < k ; ++t ) // 把排好序的部分换回去
a[l+t] = help[t] ;
}
void Merge_Sort( int l , int r ){
if( l < r ){
int mid = ( l + r ) / 2 ;
Merge_Sort( l , mid ) ;
Merge_Sort( mid+1 , r ) ;
Merge( l , mid , r ) ;
}
}
int main(){
while( ~scanf( "%d" , &n ) && n ){
int i ;
for( i = 1 ; i <= n ; ++i )
scanf( "%d" , &a[i] ) ;
ans = 0 ;
Merge_Sort( 1 , n ) ;
cout << ans << endl ;
}
return 0 ;
}
我写的这个在 POJ 只要 360 ms ,比树状数组和线段树快,时间复杂度 O(nlogn)