题目
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
解题思路
这题很复杂,以下代码是看了K神的, 这里给上我自己的解释,定义了两个成员变量nums和tmp,并且在构造函数中将传入的nums数组赋值给了成员变量nums,tmp是辅助数组。同时创建了一个tmp数组,其长度与nums数组相同,之后调用mergeSort函数进行逆序对的统计,并返回逆序对的总数。mergeSort函数的目的是将数组进行排序并统计逆序对的总数。如果数组的长度为1,则直接返回0,不需要再进行排序和统计。否则,将数组分为左右两部分,分别进行排序,并返回两部分逆序对的总数之和。在这一部分中,使用归并排序的方法进行数组的排序。定义三个指针,i、j分别指向左半部分和右半部分的第一个元素,k则用于在辅助数组tmp中重新排序并存储元素。将nums数组中的元素按照大小关系依次存入tmp数组中,再将tmp数组中的元素依次复制回nums数组,如果当前左半部分的元素值小于等于右半部分的元素值,则不统计逆序对,并将左半部分的元素存入nums数组中;否则,将右半部分的元素存入nums数组中,并统计逆序对的总数。最后返回逆序对的总数。也可以看B站视频BV1fh411r7B8讲解,具体代码如下。
Java代码实现
public class ReversePairs {
int[] nums, tmp;
public int reversePairs(int[] nums) {
this.nums = nums;
tmp = new int[nums.length];
return mergeSort(0, nums.length - 1);
}
private int mergeSort(int l, int r) {
// 终止条件
if (l >= r) return 0;
// 递归划分
int m = (l + r) / 2;
int res = mergeSort(l, m) + mergeSort(m + 1, r);
// 合并阶段
int i = l, j = m + 1;
for (int k = l; k <= r; k++)
tmp[k] = nums[k];
for (int k = l; k <= r; k++) {
if (i == m + 1)
nums[k] = tmp[j++];
else if (j == r + 1 || tmp[i] <= tmp[j])
nums[k] = tmp[i++];
else {
nums[k] = tmp[j++];
res += m - i + 1; // 统计逆序对
}
}
return res;
}
}