POJ2299 Ultra-QuickSort 归并排序求逆序数对

Problem Address：http://poj.org/problem?id=2299

 

归并排序，同时求逆序数对。

 

第一次写归并排序，其实还是蛮简单的。

 

在归并排序中加多一句话便可以同时求出逆序数对的数目。

 

思路如下：

 

假设a、b两个相邻的已排序的区间。做归并排序。c作为临时数组。

 

如果ai<=bi，说明b中剩下的数均大于ai，即ai在b剩下的数中不存在逆序数对。把ai添加到c中。

 

如果ai>bi，说明bi小于a中剩下的数，即bi均是a中剩下的数的逆序数，此时把bi添加到c中，并算出逆序数的对数=a中剩下的数。

 

对于每个ai>bi都加上相应的逆序数对数。

 

比较结束后把a或b剩下的数接到c的尾部，再把c赋到原数组上。

 

结果逆序数对数就求出来了。

 

限制： Time Limit:7000MS    Memory Limit:65536K

 

我的结果： Memory:3672K    Time:375MS

 

哈哈，不错~

 

以下贴代码：

 

#include <iostream> using namespace std; int a[500005],b[1000005]; __int64 num; void msort(int s, int t) { int mid = (t+s)/2; int qb = 1, bn = t-s+1; int i,j; if (s>=t) return; msort(s,mid); msort(mid+1,t); for (i=s,j=mid+1; i<=mid && j<=t; ) { if (a[i]<=a[j]) { b[qb] = a[i]; i++; } else { b[qb] = a[j]; num += mid-i+1; j++; } qb++; } while(j<=t) { b[qb] = a[j]; qb++; j++; } while(i<=mid) { b[qb] = a[i]; qb++; i++; } for (i=1,j=s; i<qb; i++,j++) { a[j] = b[i]; } } int main() { int n,i; while(scanf("%d", &n)!=EOF) { if (n==0) break; for (i=1; i<=n; i++) scanf("%d", &a[i]); num = 0; msort(1, n); printf("%I64d/n", num); } return 0; }