题目描述
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4]
输出: 5
思路
可以暴力,但是这道题的题解是用归并排序
代码
class Solution {
public:
int mergeSort(vector<int>& nums, vector<int>& temp, int l, int r){
if(l >= r)
return 0;
int mid = (l + r) / 2;
//将区间分为两个区间,到最后分成每一个只有一个元素的区间
int count = mergeSort(nums,temp,l,mid) + mergeSort(nums,temp,mid + 1,r);
int i = l;
int j = mid + 1;
//将子区间进行排序向上合并,同时算出这一层的逆序对数。
for(int k = l; k <= r; k++){
if(i <= mid && (j > r || nums[i] <= nums[j])) {
temp[k] = nums[i++];
count += (j - mid -1);
}else{
temp[k] = nums[j++];
}
}
for(int m = l; m <= r; m++)
nums[m] = temp[m];
return count;
}
int reversePairs(vector<int>& nums) {
int n = nums.size();
vector<int> temp(n);
return mergeSort(nums,temp,0,n-1);
}
};