(可以先去看看 leetcode 困难 —— 数组中的逆序对(分治法),几乎一样,稍微简单一点点)
题目:
给你一个整数数组 nums ,按要求返回一个新数组 counts 。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
题解:
和 数组中的逆序对 很像
一个求得是 逆序对,一个求的是 非逆序对
然后这个需要考虑位置,分治排序的时候,保存并考虑一下初始位置即可
没了
代码如下:
class Solution {
public:
vector<int> res;
vector<pair<int, int> > numbers;
void solve(int l, int r) {
int mid = (r + l) / 2;
if(r - l > 1) {
solve(l, mid);
solve(mid, r);
}
else {
return;
}
vector<pair<int, int> > temp;
int flag1 = l, flag2 = mid;
while(flag1 < mid) {
while(flag2 < r && numbers[flag1].first > numbers[flag2].first) {
temp.push_back(numbers[flag2]);
flag2++;
}
temp.push_back(numbers[flag1]);
res[numbers[flag1].second] = res[numbers[flag1].second] + flag2 - mid;
flag1++;
}
while(flag2 < r) {
temp.push_back(numbers[flag2]);
flag2++;
}
for(int i = l; i < r; i++) {
numbers[i] = temp[i - l];
}
}
vector<int> countSmaller(vector<int>& nums) {
for(int i = 0; i < nums.size(); i++) {
res.push_back(0);
numbers.push_back(make_pair(nums[i], i));
}
solve(0, nums.size());
return res;
}
};