归并排序——788：逆序对的数量

788. 逆序对的数量¹

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

输入格式

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

输出格式

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

数据范围

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

输入样例：

6
2 3 4 5 6 1

输出样例：

5

思路：

我们可以使用分治法解决问题。
我们将序列分成左右区间（本文以中间点为分界点），将逆序对分成三类：

1. 两个元素都在左边；
2. 两个元素都在右边；
3. 两个元素一个在左一个在右；

前两个好做，只需要递归算左右两边的逆序对即可。
第三个的话，需要想到一个性质：
左右两边区间排好序后对第三个的求解是不影响的。因为如果两边都是有序的，那么第三步很容易得到。不然暴力求解的话，复杂度就成了$O(n^2)$。

---

举个例子

5 7 9 | 2 4 5

当左右区间排好序后，我们只需要判断当第 $5>2$ 时，那么一定左边区间 $5$ 后面的数都大于 $2$（因为是已经排好序的 ）。这样就能减少循环计算。

代码：

# 采用归并排序模板稍作修改即可
def merge_sort(l,r):
    if l >= r:
        return 0
    mid = l + r >> 1
    i = l
    j = mid + 1
    # 用来临时保存归并好的一组数据
    tmp = []
    # 左右递归并返回逆序对个数
    num = merge_sort(l,mid) + merge_sort(mid+1,r)
    while i <= mid and j <= r:
        if data[i] <= data[j]:
            tmp.append(data[i])
            i += 1
        else:
            tmp.append(data[j])
            j += 1
            # mid - i + 1 表示从i~mid下标的所有数都和所对应的j下标的数形成逆序对
            num += mid - i + 1
    # 当一边已经结束后直接把另一边的剩余数据加入到tmp中
    tmp.extend(data[i:mid+1])
    tmp.extend(data[j:r+1])
    # 将结果返回到data中
    data[l:r+1] = tmp
    return num
n = int(input())
data = [int(x) for x in input().split()]
print(merge_sort(0,n-1))

---

1. 题目来源：AcWing ↩︎