POJ 2299 Ultra-QuickSort

本例主要是利用归并排序求逆序对数

只要在归并排序的基础上添加一行计数即可

//---------------------------------------
// 利用归并排序(MergeSort)求逆序对数
// http://baike.baidu.com/view/90797.htm
// 归并排序主要采用分治思想，分为若干子序列排序
// 时间空间复杂度和QuickSort相同，但更稳定
//---------------------------------------
#include <cstdio>
#include <cstring>
using namespace std;
#define LL long long
const int M = 500005;
LL a[M], ans, t[M];

void MergeSort(LL L, LL R)
{
    // 将序列每相邻两个数字进行归并操作（merge)，形成floor(n/2)个序列，排序后每个序列包含两个元素
    // 将上述序列再次归并，形成floor(n/4)个序列，每个序列包含四个元素
    // 重复步骤2，直到所有元素排序完毕 
    if (L == R) return;
    LL mid = (L + R) / 2;
    MergeSort(L, mid); // 排序左子区间
    MergeSort(mid + 1, R); // 排序右子区间
    LL i = L, j = mid + 1, p = 0;
    // 打印每个待排序区间中所有元素
    for (LL i = L; i <= R; i++)
        printf("%d ", a[i]);
    printf("\n");
    // 等待左右子区间合并
    while (i <= mid && j <= R)
    {
        if (a[i] > a[j])
        {
            // POJ 2299就是在归并排序基础上加这一句
            // 当a[i]>a[j]，a[i]以及以后的数都可以和a[j]形成逆序对
            ans += mid - i + 1;
            t[++p] = a[j++]; // a[j]进入合并Merge序列
        }
        else
        {
            t[++p] = a[i++];
        }
    }
    // 左右子区间剩余元素进入合并序列
    while (i <= mid) t[++p] = a[i++];
    while (j <= R) t[++p] = a[j++];
    // 讲合并序列t赋予a，当前区间结束
    p = 0;
    for (LL i = L; i <= R; i++) a[i] = t[++p];
}

int main()
{
    int n;
    while (scanf("%d", &n), n)
    {
        ans = 0;
        for (int i = 1; i <= n; i++) scanf("%lld", &a[i]);
        MergeSort(1, n);
        printf("%lld\n", ans);
    }
    return 0;
}
/*
Sample Input:
5
9 1 0 5 4
3
1 2 3
0
*/