大家好,我是何未来,本篇文章给大家讲解《AcWing算法基础课》788 题——逆序对的数量。本文详细讲解了如何通过归并排序算法高效计算数组中的逆序对数量。通过递归分治和归并过程,我们不仅实现了数组的排序,还在排序过程中巧妙地计算了逆序对的数量。文章还提供了 Java 代码实现,帮助大家更好地理解和应用这一算法。
❓题目描述
给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1 ≤ n ≤ 100000,
数列中的元素的取值范围 [1,109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
💡算法思路
- 对数组进行归并排序
- 归并排序过程中计算逆序对的数量
具体实现步骤:
- 递归分治:将数组分成两部分,分别对左半部分和右半部分进行递归排序。
- 归并过程:
- 使用两个指针
i
和j
分别指向左半部分和右半部分的当前元素。 - 比较
nums[i]
和nums[j]
,将较小的元素放入临时数组tmp
中。 - 如果
nums[i]
大于nums[j]
,计算逆序对的数量并累加到ans
中。 - 将剩余的元素依次放入
tmp
数组中。 - 将
tmp
数组中的元素复制回原数组nums
。
- 使用两个指针
看到这里可能你会产生疑问,为什么 nums[i]
大于 nums[j]
时,使用ans += mid - i + 1;
就可以计算出逆序对的数量呢?
这是因为根据归并排序的原理,每次递归会将原数组分为左右两个子数组,先对左子数组进行递归排序,再对右子数组进行递归排序,那么我们可以知道,每一次递归结束,左子数组的所有元素一定是从小到大排列的,也就是右边的元素一定比左边的元素更大。那么我们在合并两个子数组过程中当 nums[i]
大于 nums[j]
时,它们刚好组成了一个逆序对,并且左子数组中 nums[i]
后面的元素都和 nums[j]
形成了逆序对,所以就可以计算出当前递归中有多少个逆序对,把所有次递归的逆序对数量累加起来就得到了整个数组的逆序对数量。
具体判断逻辑如下:
while (i <= mid && j <= r) {
// 合并两个有序子数组
if (a[i] <= a[j]) {
// 如果左半部分的元素小于或等于右半部分的元素
tmp[k++] = a[i++]; // 将左半部分的元素放入临时数组
} else {
tmp[k++] = a[j++]; // 否则将右半部分的元素放入临时数组
// 以下为关键代码:
// 此时a[i] > a[j]&#x