题目
在数组中的两个数字如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。
输入一个数组,求出这个数组中的逆序对的总数。
样例
输入:[1,2,3,4,5,6,0]
输出:6
解法:归并排序
归并排序中,假设左右两边分别有序,然后左右两边依次比较,有序存入缓冲区,最后再复制回原数组
可以利用比较的过程计算逆序对的数目
如果左侧某个数i大于右侧某个数j,那么i后面的数自然也大于j
时间复杂度O(nlogn),空间复杂度O(n)
class Solution {
public:
int merge_sort(vector<int> &nums, int l, int r) {
if (l >= r) return 0;
int res = 0;
int mid = l + r >> 1;
res += merge_sort(nums, l, mid);
res += merge_sort(nums, mid + 1, r);
vector<int> tmp;
int i = l, j = mid + 1;
while (i <= mid && j <= r) {
if (nums[i] > nums[j]) {
res += mid - i + 1;
tmp.push_back(nums[j++]);
}
else tmp.push_back(nums[i++]);
}
while (i <= mid) tmp.push_back(nums[i++]);
while (j <= r) tmp.push_back(nums[j++]);
for (int i = l, j = 0; i <= r; i++, j++)
nums[i] = tmp[j];
return res;
}
int inversePairs(vector<int>& nums) {
return merge_sort(nums, 0, nums.size() - 1);
}
};