预备知识:归并排序
//合并两个有序数组
void merge_sort_two_vec(vector<int> &sub_vec1,vector<int> &sub_vec2,vector<int> &vec) {
int i = 0,j = 0;
while(i < sub_vec1.size() && j < sub_vec2.size()) {
if(sub_vec1[i] < sub_vec[j]) {
vec.push_back(sub_vec[i]);
i++;
}
else {
vec.push_back(sub_vec[j]);
j++;
}
}
//如果sub_vec1还有剩余元素
for(; i < sub_vec1.size(); i++) {
vec.push_back(sub_vec[i]);
}
//如果sub_vec2还有剩余元素
for(; j < sub_vec2.size(); j++) {
vec.push_back(sub_vec[j]);
}
}
void merge_sort(vector<int> &vec) {
if(vec.size() < 2) { //子问题足够小,直接返回
return;
}
int mid = vec.size()/2;
vector<int> sub_vec1;
vector<int> sub_vec2;
for(int i = 0; i < mid; i++) {
sub_vec1.push_back(vec[i]);
}
for(int j = mid; j < vec.size(); j++) {
sub_vec2.push_back(vec[j]);
}
merge_sort(sub_vec1);
merge_sort(sub_vec2);
vec.clear();
//合并,将子问题的解进行合并
merge_sort_two_vec(sub_vec1,sub_vec2,vec);
}
给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
示例:
输入:nums = [5,2,6,1]
输出:[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素
class Solution {
public:
vector<int> countSmaller(vector<int>& nums) {
//TODO
}
};