题目链接:leetcode315
给你一个数组,让你求每个位置的数后面比它小的数的数量。也就是以每个数字为头的逆序对的数量。
解题思路
树状数组、线段树
先离散化一下,然后逆序统计每个位置前比该数字小的数。
分治
先离散化,再利用归并排序把数组逆序,并求后面的区间对前面区间的贡献。
代码
分治
class Solution {
public:
vector<int> countSmaller(vector<int>& nums) {
int n = nums.size();
vector<int> countArray(n, 0);
vector<pair<int, int> > arr;
for (int i = 0; i < n; i++) arr.push_back({nums[i], i}); //离散化
MergeSort(0, n-1, arr, countArray);
return countArray;
}
void MergeSort(int l, int r, vector<pair<int, int> >& arr, vector<int>& countArray) {
if (l >= r) return ;
int mid = l + (r - l >> 1);
MergeSort(l, mid, arr, countArray);
MergeSort(mid + 1, r, arr, countArray);
int i = l, j = mid + 1;
vector<pair<int, int> > tmp;
while (i <= mid && j <= r) {
if (arr[i].first > arr[j].first) {
tmp.push_back(arr[i]);
countArray[arr[i].second] += r - j + 1;
i++;
} else {
tmp.push_back(arr[j]);
j++;
}
}
while (i <= mid) tmp.push_back(arr[i++]);
while (j <= r) tmp.push_back(arr[j++]);
for (i = l; i <= r; i++) arr[i] = tmp[i-l];
}
};