AcWing:788. 逆序对的数量

题目

给定一个长度为 n 的整数数列，请你计算数列中的逆序对的数量。

逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i<j 且 a[i]>a[j]，则其为一个逆序对；否则不是。

输入格式

第一行包含整数 n，表示数列的长度。

第二行包含 n 个整数，表示整个数列。

输出格式

输出一个整数，表示逆序对的个数。

数据范围

1≤n≤100000，
数列中的元素的取值范围 [1,1e9]。

输入样例：

6
2 3 4 5 6 1

输出样例：

5

思路简介 归并排序 + 计数统计

思路简介
利用归并排序, 在排序过程中, 记录两个区间前后构成的逆序对, 并返回该值
再递归的过程中, 最终会递归到两个区间都只有一个数的情况, 这试,只需记录前后逆序对的数量(1 或 0), 返回他的上级调用
这是两个递归区间都有 2 个数, 但这两个区间 res(逆序对数) 已经累加上次调用的逆序对数量吗, 再加上这两个区间的逆序对数量, 返回上级调用
以此类推, 最终答案就产生了
中间排序是不会打乱每个区间的相对位置的, 故不会逆序对的计数产生影响
总结: 思路为归并排序 + 计数统计

• 注意 结果 res 需要用 long long 存储

#include <iostream>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int q[N], tmp[N];
int n;

LL merge_sort(int l, int r) {
    if (l >= r) return 0;
    int mid = l + r >> 1;
    LL res = merge_sort(l, mid) + merge_sort(mid + 1, r);  // 递归求出之前的逆序对的个数
    // 排序
    int k = 0, i = l, j = mid + 1; 
    while (i <= mid && j <= r) {
        if(q[i] <= q[j]) tmp[k ++] = q[i ++];
        else {                  // 由于正在排序的两个区间都是有序的(升序) ,
            res += mid - i + 1; // 当 q[i] < q[j] 时, q[i ~ mid] 的所有数, 都可以与 q[j]构成逆序对 
            tmp[k ++] = q[j ++];
        }
    }
    // 收尾
    while (i <= mid) tmp[k ++] = q[i ++];
    while (j <= r) tmp[k ++] = q[j ++];
    // 复位
    for (int i = l, j = 0; i <= r; i ++, j ++) q[i] = tmp[j];
    // 返回当前两个区间逆序对的数量, 此时当前大区间已有序
    return res;
}


int main() {
    cin >> n;
    for (int i = 0; i < n; i ++) cin >> q[i];
    cout << merge_sort(0, n - 1);
    return 0;
}