通过归并排序统计逆序对的方法
左半部分 [l, mid] 已经是递增序列
右半部分 [mid+1, r]也是递增序列
但是当它们合并在一起时可能会形成逆序对,i在左半部分,j在右半部分。
nums[j] < nums[i]时,nums[j]肯定大于nums[i]之前的所有数(否则早就加到tmp里了,也不会等到现在)。这个时候意味着,nums[i + 1], nums[i + 2], nums[i + 3]......都大于nums[j](因为左半部分递增,nums[i]你都不大于,剩下的咋可能大于),但是同时nums[j]是比右半部分的nums[j - 1], nums[j - 2]...那些大(也因为右半部分递增)。所以逆序对的数量就是 i 到 mid 中间那些大于j的数,包括i和mid,所以是i - mid + 1。
动笔画一下就明白了。
#include <vector>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型vector
* @return int整型
*/
const int MOD = 1e9 + 7;
vector<int> tmp;
long long mergesort(vector<int>& nums, int l, int r) {
if(l >= r) return 0;
int mid = l + r >> 1;
long long res = mergesort(nums, l, mid) + mergesort(nums, mid + 1, r);
int k = 0, i = l, j = mid + 1;
while(i <= mid && j <= r) {
if(nums[i] <= nums[j]) tmp[k ++] = nums[i ++];
else {
tmp[k ++] = nums[j ++];
res = (res + mid - i + 1) % MOD;
}
}
while(i <= mid) tmp[k ++] = nums[i ++];
while(j <= r) tmp[k ++] = nums[j ++];
for(int i = l, j = 0; i <= r; i ++, j ++) nums[i] = tmp[j];
return res;
}
int InversePairs(vector<int>& nums) {
tmp.resize(nums.size());
return mergesort(nums, 0, nums.size() - 1);
}
};

京公网安备 11010502036488号