[归并排序] 洛谷P1908 逆序对 （分治应用）

题目

题目描述

猫猫TOM和小老鼠JERRY最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。最近，TOM老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中ai>aj且i < j的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。
输入输出格式
输入格式：
第一行，一个数n，表示序列中有n个数。
第二行n个数，表示给定的序列。
输出格式：
给定序列中逆序对的数目。
输入样例#1:
6
5 4 2 6 3 1

输出样例#1:
11

说明
对于50%的数据，n≤2500
对于100%的数据，n≤40000。

代码

#include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std;

const int maxn = 50000;
int cnt = 0;
int T[maxn], A[maxn];

void solve(int *A, int x, int y, int *T) {
    if (y - x > 1) {
        int m = x + (y - x) / 2;
        solve(A, x, m, T);
        solve(A, m, y, T);
        int p = x, q = m, i = x;
        while (p < m || q < y) {
            if (q >= y || (p < m && A[p] <= A[q])) T[i++] = A[p++];
            else { T[i++] = A[q++]; cnt +=m-p; }
        }
        for (int i = x; i < y; i++) A[i] = T[i];
    }
}

int main() {
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++) scanf("%d", &A[i]);
    solve(A, 0, n, T);

    printf("%d\n", cnt);

    return 0;
}

思路

1.分治三步走

• 划分问题：把序列分成尽量相等的两半分别求解
• 递归求解：那不就递归求解嘛
• 合并自问题：本题难点。可以对于右边的每个j，统计左边比它大的元素个数f(j)，则所有f(j)之和就是答案。而本题之所以与归并排序有关，就是因为归并排序正好顺便完成了f(j)的计算：由于归并排序的合并是从小到大进行的，那么当右边的A[j]复制到T时，左边的还没来得及复制到T的所有元素的个数就是f(j)（因为它们都比A[j]大）。具体的代码操作就是在累加器上加上左边元素个数m-p即可。