在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4]
输出: 5
限制:
0 <= 数组长度 <= 50000
题解:最开始想用DP,写着写着发现变成了暴力,简单分析了一下因为有后效性,所以用不了DP。看了题解知道原来使用的是并归排序,简单来说就是在对某两段数组进行并归的时候计算逆序数对,左边数组中的第i个数字如果比右边数组中的第j个数字大的话,这个i到左边数组的结尾处的距离就是这个数字与右边数组能够组成的逆序数对的数量。
PS 需要注意的是,自己在写的时候在想,会不会出现两段数组在排序前和排序后有不一样的逆序数对的情况,其实不会。相当于是其中一个数组中的一个数字与另一个数组中的数字组成逆序数对的问题,与排序位置无关。各子数组中的逆序数对已经算好了。
class Solution {
public:
int ans;
vector<int> v;
void MergeSort(vector<int>& nums, int left, int right) {
if (left < right) {
int mid = (left+right) / 2;
MergeSort(nums, left, mid);
MergeSort(nums, mid+1, right);
Merge(nums, left, mid, right);
}
}
void Merge(vector<int>& nums, int left, int mid, int right) {
int i = left;
int j = mid + 1;
int pos = left;
while (i <= mid && j <= right) {
if (nums[i] > nums[j]) {
v[pos++] = nums[j++];
ans += (mid - i + 1);
}
else v[pos++] = nums[i++];
}
while (i <= mid) v[pos++] = nums[i++];
while (j <= right) v[pos++] = nums[j++];
for (int k = left; k <= right; ++k) nums[k] = v[k];
//cout << ans << endl;
}
int reversePairs(vector<int>& nums) {
ans = 0;
if (nums.size() == 0 || nums.size() == 1) return ans;
int n = nums.size();
v.assign(nums.begin(), nums.end());
//for (int i = 0; i < n; ++i) cout << v[i] << " ";
MergeSort(nums, 0, n-1);
return ans;
}
};