Count of Smaller Numbers After Self | LeetCode
题意理解
我一开始看到这个题的时候,觉得挺简单,至少题意很容易理解,使用O(N*N)的解法肯定能够得出答案,只是提交的时候回出现超时。想了一下,感觉应该还有地方可以优化,总觉得应该可以使用排序来辅助。分析下题意,每个位置的结果应该是这个数后面小于这个数的个数,构建一个新数组,然后将原数组反向插入并排序,每个数第一次被插入的位置就是这个数在原数组中的结果,然后反转,就可以得到答案,代码如下:
代码
vector<int> countSmaller(vector<int>& nums) {
int size = nums.size();
vector<int> res;
vector<int> temp;
for(int i=size-1;i>=0;--i){
res.push_back(insert(temp,nums[i]));
}
reverse(res.begin(),res.end());
return res;
}
int insert(vector<int>& temp,int num){
int size = temp.size();
int res = distance(temp.begin(),lower_bound(temp.begin(),temp.end(),num));
temp.insert(temp.begin()+res,num);
return res;
}