难度:困难
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
输入: [7,5,6,4]
输出: 5
限制:
0 <= 数组长度 <= 50000
预备知识
「归并排序」是用分治思想,分治模式在每一层递归上有三个步骤:
n
个元素分成个含 n/2
个元素的子序列。
具体的我们以一组无序数列{14,12,15,13,11,16}
为例分解说明,如下图所示:
在待排序序列长度为 1
的时候,递归开始「回升」,因为我们默认长度为 1
的序列是排好序的。
具体思路:
那么求逆序对和归并排序又有什么关系呢?关键就在于「归并」当中「并」的过程。
合并阶段 本质上是 合并两个排序数组 的过程:
左子数组当前元素
>
右子数组当前元素
时,意味着
左子数组当前元素i 至 末尾元素m
」 与 「右子数组当前元素
」 构成了若干 「逆序对
」 ;cnts
+= (m - i + 1)
。C++
class Solution {
private:
int cnts = 0;
void mergeSort(vector<int>& nums, vector<int>& tmp, int l, int h){
if(h - l < 1) return;
//分解
int m = l + (h - l) / 2;
mergeSort(nums, tmp, l, m);
mergeSort(nums, tmp, m + 1, h);
//解决 + 合并
int k = l, i = l, j = m + 1;
while(i <= m || j <= h){
if(i > m) tmp[k++] = nums[j++];
else if(j > h || nums[i] <= nums[j]) tmp[k++] = nums[i++];
else{//此时i~m对应数组中的数都比nums[j]大
tmp[k++] = nums[j++];
cnts += (m - i + 1);
}
}
for(k = l; k <= h; k++){
nums[k] = tmp[k];
}
}
public:
int reversePairs(vector<int>& nums) {
vector<int> tmp(nums.size());//辅助数组,临时记录中间合并的子数组
mergeSort(nums, tmp, 0, nums.size() - 1);
return cnts;
}
};
Java
class Solution {
private int cnts = 0;
private void mergeSort(int[] nums, int[] tmp, int l, int h){
if(h - l < 1) return;
//分解
int m = l + (h - l) / 2;
mergeSort(nums, tmp, l, m);
mergeSort(nums, tmp, m + 1, h);
//解决 + 合并
int k = l, i = l, j = m + 1;
while(i <= m || j <= h){
if(i > m) tmp[k++] = nums[j++];
else if(j > h || nums[i] <= nums[j]) tmp[k++] = nums[i++];
else{//此时i~m对应数组中的数都比nums[j]大
tmp[k++] = nums[j++];
cnts += (m - i + 1);
}
}
for(k = l; k <= h; k++){
nums[k] = tmp[k];
}
}
public int reversePairs(int[] nums) {
int[] tmp = new int[nums.length];//辅助数组,临时记录中间合并的子数组
mergeSort(nums, tmp, 0, nums.length - 1);
return cnts;
}
}
n
为数组的长度,同归并排序 O ( n l o g n ) O(nlogn) O(nlogn)。题目来源:力扣。
放弃一件事很容易,每天能坚持一件事一定很酷,一起每日一题吧!
关注我LeetCode主页 / CSDN—力扣专栏,每日更新!