LeetCode315 计算右侧小于当前元素的个数

添加链接描述## 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/