添加链接描述## LeetCode315 计算右侧小于当前元素的个数
题目描述:给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
示例:
输入:[5,2,6,1]
输出:[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
vector<int> countSmaller(vector<int>& nums)
{
vector<pair<int,int>> nums_pair;//将nums下标和元素值对应起来
vector<int> count;//作为结果数组返回
for(int i = 0; i < nums.size(); i++)
{
nums_pair.push_back(make_pair(nums[i],i));
count.push_back(0);
}
merge_sort(nums_pair, count);
return count;
}
void merge_sort(vector<pair<int,int>>& vec, vector<int>& count)
{
if(vec.size() < 2)
{
return;//容量为1,直接return
}
int mid = vec.size() / 2;//分解成子问题
vector<pair<int,int>> vec1;
vector<pair<int,int>> vec2;
for (int i = 0; i < mid ; ++i)
{
vec1.push_back(vec[i]);
}
for (int j = mid; j < vec.size(); ++j)
{
vec2.push_back(vec[j]);
}
merge_sort(vec1, count);
merge_sort(vec2, count);//对子问题再进行分解
vec.clear();//清空原数组
merge_sort_two_vector(vec1, vec2, vec, count);//进行归并排序
}
void merge_sort_two_vector(vector<pair<int,int>>& nums1, vector<pair<int,int>>& nums2,vector<pair<int,int>>& vec, vector<int>& count)
{
int i = 0;
int j = 0;
while( i < nums1.size() && j < nums2.size())
{
if(nums1[i].first <= nums2[j].first)//
{
count[nums1[i].second] += j;//每当对该值进行一次排序,意味着即有j个值小于它
vec.push_back(nums1[i]);
++i;
} else{
vec.push_back(nums2[j]);
++j;
}
}
for(; i < nums1.size(); ++i)
{
//同上
count[nums1[i].second] += j;
vec.push_back(nums1[i]);
}
for(; j < nums2.size(); ++j)
{
vec.push_back(nums2[j]);
}
}
};
int main()
{
// vector<int> vec1 = {1,3,5,7,9};
// vector<int> vec2 = {2,4,6,8,10};
vector<int> vec;
int test[] = {5,2,6,1};
for(int i = 0; i < 4; ++i)
{
vec.push_back(test[i]);
}
vector<int> res = Solution().countSmaller(vec);
for (int j = 0; j < res.size(); ++j)
{
printf("%d,", res[j]);
}
return 0;
}
题目链接:
https://leetcode-cn.com/problems/count-of-smaller-numbers-after-self/