问题定义
给定一个长度为n的整数数组,计算数组中的逆序对的数量并返回
逆序对的定义如下:对于数组的第i
个和第j
个元素,如果满足i
<j
且a[i]
>a[j]
,则其为一个逆序对;否则不是。
数据范围
1≤n≤100000,数组中的元素的取值范围 [1,1e9]。
示例:输入[2,3,4,5,6,1]
,输出5
解释:逆序对共5对,[2, 1], [3, 1], [4, 1], [5, 1], [6, 1]
本题暴力硬解的复杂度为$O(n^2)$,较为繁琐,于是考虑是否存在复杂度为$O(nlogn)$的算法
要求逆序数,就不能在寻找数对时未经相关处理就更改数组元素顺序打乱原有顺序,因此要找到一种在实施过程中能够在一定程度上保留元素顺序关系的算法。
设想一种很理想的情况:元素a
与b
满足逆序数的定义(即a>b
且i<j
),同时还已知有一批比a
大的数,那么很容易推知这一批数都可以与b
构成逆序数,其个数就等于这批数的个数
以上情况正好符合归并排序的过程,首先在这里复习一下归并排序的原理:
归并排序
归并排序先递归地两两划分数组,每轮递归中都将整个数组分为左右两个部分,然后在每一次左右划分完成后,为左右两个数组各设置一个指针i
与j
,比较大小并将小者放入一个临时数组中(这个临时数组的长度是本轮划分左右数组的总长),放入一边的指针前进一步,再继续比较。当某一边全部比较完后,将可能还未放完的另一边数组元素全部放入临时数组
代码可以是:
class Solution {
public int[] mergeSort(int[] nums, int l, int r) {
// l=r证明已经划分到单个元素,返回的排序数组就是以这个单元素组成的数组
if (l == r) return new int[]{nums[l]};
int mid = (l + r) / 2;
int[] leftNums = mergeSort(nums, l, mid);
int[] rightNums = mergeSort(nums, mid+1, r);
// 能执行到这里,说明左右划分都已完成,此时如果左右都是单元素数组[2][3],则这个栈会返回[2,3]
// 此时如果左右都是多元素数组,那至少能保证这两个数组各自的内部是有序的
int[] tempNums = new int[leftNums.length + rightNums.length];
int i = 0, j = 0, m = 0; // i左j右,m是临时数组索引
while (i < leftNums.length && j < rightNums.length)
tempNums[m++] = leftNums[i] < rightNums[j] ? leftNums[i++] : rightNums[j++];
// 接下来处理一边走完一边还没走完的情况,由于单边必定已经有序,所以直接顺序放入即可
while (i < leftNums.length) tempNums[m++] = leftNums[i++];
while (j < rightNums.length) tempNums[m++] = rightNums[j++];
return tempNums;
}
}
题解
回到本题,核心在于:在归并排序比较leftNums[i]
与rightNums[j]
的大小关系时,如果正好左大于右,(即满足逆序数的定义),则可以直接确定这两个数之间的数都满足逆序数的定义
例如,
[2,4,6,7]
和[1,3,5,6]
之间,比较2
与1
发现2>1,那么左侧数组中包括2
在内的剩余元素都与1
形成逆序数关系,即[2,1],[4,1],[6,1],[7,1]
继续操作,将
1
放入临时数组,j
指向3
,2<3,将2
放入临时数组,i
指向4
此时再次发生左大于右即4>3,则又组成了一批逆序数:
[4,3],[6,3],[7,3]
如上过程,不断循环
要想方便地得到它们之间的逆序数个数,可以从索引下手。在这样的考量下,归并排序时的传参就一定要是完整数组,每次递归处理完成后都要将临时数组的元素转移到nums
中,在之后使用时也要用绝对索引值去定位元素,而非子数组的从索引0出发
整个代码只需要对归并排序做很少的改动:
class Solution {
private count = 0;
public void findReverse(int[] nums, int l, int r) {
if (l == r) return;
int mid = (l + r) / 2;
findReverse(nums, l, mid);
findReverse(nums, mid+1, r);
int[] tempNums = new int[r-l+1];
int i = l, j = mid+1, m = 0;
while (i <= mid && j <= r) {
if (nums[i] > nums[j]) {
this.count += mid - i + 1; // 可以认为只有这里有改动,即增加了逆序数的计数操作
tempNums[m++] = nums[j++];
} else {
tempNums[m++] = nums[i++];
}
}
while (i <= mid) tempNums[m++] = nums[i++];
while (j <= r) tempNums[m++] = nums[j++];
for (i = l, j = 0; i <= r; i++, j++)
nums[i] = tempNums[j];
}
}