315. Count of Smaller Numbers After Self

这道题是给出一个无序数组,返回一个counts数组,使得counts[i]是nums[i]右边比他小的数的个数。

一开始的思路是错的,以为用动态规划,从后往前算counts[i],认为在计算counts[i]时,看i的右边,找到第一个比他小的数,+1就是counts[i]。但是提交了发现整个思路是错的。因为数组不是有序的,所以比如2,0,1这个数组,这样的想法就是错的,2后面遇到的第一个比2小的数是0,所以计算成了1,但实际上1也比2小。

这道题应该用二分查找,新建一个数组tmp,从后往前遍历nums,遇到nums[i]把他用二分查找的方法插入数组tmp,插入的位置就是nums[i]后面比他小的数的个数。可以手写二分查找,也可以用lower_bound和distance函数。原理是一样的。手写二分查找的时候,因为在插入的时候可以用insert函数,insert函数是插入在给定位置的前面,所以在二分查找设置left和right的时候,令right是最右边不满足条件的值,left是满足条件的值。也就是初始化的时候令left = 0,right = tmp.size()。更新他们的时候,right = mid或left = mid + 1;

class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
        vector<int> res(nums.size(), 0);
        vector<int> tmp;
        for(int i = nums.size()-1; i >= 0; --i){
            int left = 0, right = tmp.size();
            while (left < right) {
                int mid = left + (right - left) / 2;
                if (tmp[mid] >= nums[i]) right = mid;
                else left = mid + 1;
            }
            res[i] = right;
            tmp.insert(tmp.begin() + right, nums[i]);
        }
        return res;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值