题目链接如下:
Leetcode 493-翻转对
剑指Offer 51-数组中的逆序对
两个题目类似,先看493题:
暴力解法即进行两层遍历,时间复杂度为O(n^2),这显然不符合要求。为了实现更低的时间复杂度,使用递归的归并排序。
思路:
①求出中间位置mid
,对 [left, mid] 、[mid+1, right]
两个区间分别做归并排序,并返回左右两个区间的逆序对数。
②所求 = 左区间逆序对数 + 右区间逆序对数 + 左右之间的逆序对数。
③对两个区间进行合并,并返回上一层递归。
递归结束的条件:当left == right时,只剩1个元素无需排序,翻转对为0
为什么排序:
①左右区间内各自排序不影响分别位于左右区间两数的前后相对关系;
②排序后对于统计“重要翻转对”更方便:对于某个左区间的nums[i]
,从mid+1
开始向右遍历,找到第一个不符合(nums[i] <= 2*nums[j]
)翻转对定义的nums[j]
,由于排序后是递增的,可以保证从[mid+1, j-1]
都是可以和nums[i]
组成重要翻转对的。
归并排序后、统计区间之间翻转对数并返回,有点像Leetcode 124-二叉树中的最大路径和的解法,都是既给递归的上一层返回了值(那个题是返回该节点左右子树两个分支中的更长的路径,因为返回上层的不能分叉;本题是返回本层递归的逆序对数),同时进行了某些操作(那个题是更新最大值,本题是进行排序)。
代码如下:
class Solution {
public:
int reversePairs(vector<int>& nums) {
if(nums.size()==0)
return 0;
return mergeSort(nums,0,nums.size()-1);
}
int mergeSort(vector<int>& nums,int l,int r){
if(l==r)
return 0