给定一个数组 nums ,如果 i < j 且 nums[i] > 2*nums[j] 我们就将 (i, j) 称作一个重要翻转对。
你需要返回给定数组中的重要翻转对的数量。
示例 1:
输入: [1,3,2,3,1]
输出: 2
示例 2:
输入: [2,4,3,5,1]
输出: 3
注意:
给定数组的长度不会超过50000。
输入数组中的所有数字都在32位整数的表示范围内。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-pairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
大半夜看到这道题心态崩了,hh,这就就把官方的题解放上来吧,难度确实不小,等有时间好好思考一波,下面是参考代码:
class Solution {
public:
vector<int> tmp;
int reversePairs(vector<int>& nums) {
int n = nums.size();
tmp.resize(n);
return merge_sort(nums, 0, n);
}
int merge_sort(vector<int>& nums, int l, int r) {
if (r - l <= 1) return 0;
int mid = l + (r - l) / 2;
int count = merge_sort(nums, l, mid) + merge_sort(nums, mid, r);
int j = mid;
for (int i = l; i < mid; ++i) {
while (j < r && nums[i] / 2.0 > (double)nums[j]) ++j;
count += j - mid;
}
int pos = l, l1 = l, l2 = mid;
while (l1 < mid && l2 < r) {
if (nums[l1] < nums[l2]) {
tmp[pos++] = nums[l1++];
}else tmp[pos++] = nums[l2++];
}
while (l1 < mid) tmp[pos++] = nums[l1++];
while (l2 < r) tmp[pos++] = nums[l2++];
for (int i = l; i < r; ++i) nums[i] = tmp[i];
return count;
}
};