leetcode 困难 —— 计算右侧小于当前元素的个数(分治法)

(可以先去看看 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;
    }
};
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值