归并排序求逆序对的数量

合集 - 1-1.基础算法(21)

1.论读书2023-10-252.快速排序2023-10-053.快速选择2023-10-064.归并排序2023-10-10

5.归并排序求逆序对的数量2023-10-11

6.二分查找（整数二分）2023-10-117.二分查找（浮点二分）2023-10-128.高精度加法2023-10-139.高精度减法2023-10-1310.高精度乘法2023-10-1511.高精度除法2023-10-1612.前缀和（一维）2023-10-1713.子矩阵的和（二维前缀和）2023-10-1814.差分（一维）2023-10-1915.差分矩阵（二维差分）2023-10-2016.最长连续不重复子序列（双指针）2023-11-2117.数组元素的目标和（双指针）2023-11-2118.判断子序列（双指针）2023-11-2219.位运算2023-11-2220.离散化2023-11-2221.区间合并2023-11-22

收起

一、题目来源

AcWing算法基础课-788.逆序对的数量

二、题目描述

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

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

输入格式**

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

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

输出格式

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

数据范围

$1\le n\le 100000，$
数列中的元素的取值范围 $[1,{10}^9]$。

输入样例：

6
2 3 4 5 6 1 

输出样例：

5 

三、算法思路

本题基于归并排序的过程。

思路如下：

1. 先将区间分为两段

2. 满足逆序对条件的两个点有如下三种情况：

   1. 两个点都在左区间
   2. 两个点都在右区间
   3. 一个点在左边区间，一个点在右边区间

3. 不断的递归排序的过程中，上述 $1、2$ 情况都会转化为 $3$ 的情况，所以答案直接加上左区间排序和右区间排序即可

4. 由于归并之后两个区间都是有序的，所以如果当前左区间的数大于当前右区间的数，则左区间剩下的所有数都大于当前右区间的数，即满足逆序对的条件，res += mid - i + 1;

• 如果直接暴力解题，从后往前遍历，时间复杂度$O(n^2)$，采用归并排序解题时间复杂度为$O(nlogn)$。
• 注意思考上述三种情况，在不断划分区间的过程中，最后都会转化为第三种情况。
• 数据量很大可能会爆int，所以用long long来存答案。

四、源代码

#include <iostream>

using namespace std;

const int N = 100010;

typedef long long LL;

int n;
int a[N];
int tmp[N];

LL merge_sort(int a[], int l, int r)
{
    if (l >= r) return 0;
    
    int mid = (l + r) >> 1;
    LL res = merge_sort(a, l, mid) + merge_sort(a, mid + 1, r);
    
    int k = 0, i = l, j = mid + 1;
    while (i <= mid && j <= r)
        if (a[i] <= a[j])   tmp[k ++ ] = a[i ++ ];
        else    res += mid - i + 1, tmp[k ++ ] = a[j ++ ];
    while (i <= mid)    tmp[k ++ ] = a[i ++ ];
    while (j <= r)  tmp[k ++ ] = a[j ++ ];
    
    for (int i = l, j = 0; i <= r; ++i, ++j)    a[i] = tmp[j];
    
    return res;
}

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