题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4]
输出: 5
限制:0 <= 数组长度 <= 50000
参考解题思路:归并排序,分而治之
- 分: 不断将数组从中点位置划分开(即二分法),将整个数组的排序问题转化为子数组的排序问题;
- 治: 划分到子数组长度为 1 时,开始向上合并,不断将 较短排序数组 合并为 较长排序数组,直至合并至原数组时完成排序;
算法流程:
reverseNums() 归并排序与逆序对统计:
- 终止条件: 当 left≥right 时,代表子数组长度为 1 ,此时终止划分;
- 递归划分: 计算数组中点 middle ,递归划分左子数组 reverseNums(left, middle) 和右子数组 reverseNums(middle + 1, right) ;
- 合并与逆序对统计:
- 暂存数组 num闭区间 [i, right] 内的元素至辅助数组tmp ;
- 循环合并: 设置双指针 i , j分别指向左 / 右子数组的首元素;
- 当 i = middle + 1 时: 代表左子数组已合并完,因此添加右子数组当前元素 tmp[j] ,并执行 j = j + 1 ;
- 当 j = r + 1 时: 代表右子数组已合并完,因此添加左子数组当前元素tmp[i] ,并执行 i = i + 1 ;
- 当tmp[i] ≤ tmp[j] 时: 添加左子数组当前元素tmp[i] ,并执行 i = i + 1;
- 当(即 tmp[i] > tmp[j])时: 添加右子数组当前元素 tmp[j],并执行 j = j + 1;此时构成 m - i + 1 个「逆序对」,统计添加至 res ;
- 返回值: 返回直至目前的逆序对总数 res ;
reversePairs() 主函数:
- 初始化: 辅助数组 tmp ,用于合并阶段暂存元素;
- 返回值: 执行归并排序 reverseNums() ,并返回逆序对总数即可;
int[] num, temp;
public int reversePairs(int[] nums) {
this.num = nums;
this.temp = new int[nums.length];
return reverseNums(0, nums.length - 1);
}
public int reverseNums(int left, int right){
// 边界处理
if (left >= right) {
return 0;
}
// 分治
int middle = (left + right) / 2;
int res = reverseNums(left, middle) + reverseNums(middle + 1, right);
// 合并
int i = left, j = middle + 1;
for (int k = left; k <= right; k++) {
temp[k] = num[k];
}
for (int k = left; k <= right; k++) {
if (i == middle + 1) {
// 左子数组全部处理完--即只需要放入右子数组数据
num[k] = temp[j++];
} else if (j == right + 1 || temp[i] <= temp[j] ) {
// 右子数组全部处理完or左子数组元素小于右子数组元素--即放入左子数组元素即可
num[k] = temp[i++];
} else {
// 左子数组比右子数组大--即添加逆序数量为m - i + 1
num[k] = temp[j++];
res += middle - i + 1;
}
}
return res;
}
复杂度分析:
时间复杂度 O(NlogN) : 其中N为数组长度;归并排序使用 O(NlogN) 时间;
空间复杂度O(N) : 辅助数组tmp占用 O(N)大小的额外空间;
作者:jyd
链接:https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/solution/jian-zhi-offer-51-shu-zu-zhong-de-ni-xu-pvn2h/