这道题是给出一个无序数组,返回一个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;
}
};