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

给定一个整数数组 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 个更小的元素.

class Solution {

public:

    vector countSmaller(vector& nums) {

        vector> vec;

        vector count;

        for(int i = 0;i < nums.size();i++)

        {

            vec.push_back(make_pair(nums[i],i));

            count.push_back(0);

        }

        merget_sort(vec,count);

        return count;

    }

    void merget_sort(vector>& vec,vector& count)

    {

        if(vec.size() < 2)

            return;

        int mid = vec.size()/2;

        vector> sub_vec1;

        vector> sub_vec2;

        for(int i = 0;i < mid;i++)

            sub_vec1.push_back(vec[i]);

        for(int i = mid;i < vec.size();i++)

            sub_vec2.push_back(vec[i]);

        merget_sort(sub_vec1,count);

        merget_sort(sub_vec2,count);

        vec.clear();

        merget_sort_two_vec(sub_vec1,sub_vec2,vec,count);

    }

    void merget_sort_two_vec(vector>& sub_vec1,vector>& sub_vec2,vector>& vec,vector& count)

    {

        int i = 0;

        int j = 0;

        while(i < sub_vec1.size() && j < sub_vec2.size())

        {

            if(sub_vec1[i].first <= sub_vec2[j].first)

            {

                count[sub_vec1[i].second] += j;

                vec.push_back(sub_vec1[i]);

                i++;

            }

            else

            {

                vec.push_back(sub_vec2[j]);

                j++;

            }

        }

        for(;i < sub_vec1.size();i++)

        {

            count[sub_vec1[i].second] += j;

            vec.push_back(sub_vec1[i]);

        }

        for(;j < sub_vec2.size();j++)

        {

            vec.push_back(sub_vec2[j]);

        }

    }

};