AcWing 788.逆序对数量（归并 + 分治最详解）

techpupil

已于 2024-01-26 00:56:06 修改

阅读量132

点赞数 1

文章标签：算法数据结构

于 2024-01-25 23:57:17 首次发布

本文链接：https://blog.csdn.net/nuc_ghp/article/details/135856361

版权

[题目概述]

给定一个长度为 n 的整数数列，请你计算数列中的逆序对的数量。
逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i<j 且 a[i]>a[j]，则其为一个逆序对；否则不是。

输入格式

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

输出格式

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

数据范围

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

输入样例：

6
2 3 4 5 6 1

输出样例：

分析问题
1. 本题首先的想法就是暴力，可以过差不多80%的数据，并且如果只是考场上的题，我们应该也会选择暴力。但平时练习中，我们还是要尽量AC。
2. 本题正确的做法是归并分治（说实话根本想不到是用归并来解这道题，y总还是yyds的）。我们可以将整个区间一分为二，那么逆序对的出现就有三种情况，两个数都在左边，两个数都在右边，两个数一左一右。
  
  假设merge_sort( l, r ) 函数可以返回区间中的逆序对数量，那么第一、二中情况就可以直接返回
  merge_sort( l, mid) + merge_sort( mid + 1, r ),而第三种情况就需要思考一下。
3. 当左右两遍都排序好后，设置 i 和 j ，分别指向左右部分的端点，如果num[ i ] > num[ j ], 则符合逆序对的条件，并且 i 后面的数据也一定大于 num[j], 前面的数一定小于num[ j ]（如果小于的话就不会移动 i ，而是移动 j 了，i 在当前位置就说明前面的数都小于了）。所以此时的 i 所有的逆序对数量为 mid - i + 1。只需在归并排序中加上这句话即可。
4. 细节问题：我们最大答案的情况是所有数据降序排列，则逆序对数量为 $n - 1 + n - 2 + ... + 0 = n * (n - 1) /2$ ,最大情况是 $5 * 10^9$ ，所以会爆int，需要用long long 来存答案
完整代码

#include <iostream>
using namespace std;
const int N = 1000005;
int n;
int num[N], temp[N];
long long  cnt = 0;
long long merge_sort(int l, int r){
    if(l >= r) return 0;
    int mid = (l + r) / 2;
    cnt = merge_sort(l, mid) + merge_sort(mid + 1, r);
    
    // 归并排序
    int k = 0, i = l, j = mid + 1;
    while(i <= mid && j <= r){
        if(num[i] <= num[j]) temp[k ++] = num[i ++];
        else{
            temp[k ++] = num[j ++];
            cnt += mid - i + 1;
        }
    }
    // 处理遗留的数据
    while(i <= mid) temp[k ++] = num[i ++];
    while(j <= r) temp[k ++] = num[j ++];
    
    // 将中间数组拷贝回原数组
    for(int i = l, j = 0; i <= r; i ++, j ++) num[i] = temp[j];
    return cnt;
}
int main (){
    cin >> n;
    for(int i = 0; i < n; i ++) cin >> num[i];
    cout << merge_sort(0, n - 1) << endl;
    return 0;
}