c++用归并排序计算右侧小于当前元素个数

最新推荐文章于 2022-08-06 23:41:42 发布

前行_的路上

最新推荐文章于 2022-08-06 23:41:42 发布

阅读量132

点赞数

分类专栏： c++ 文章标签：散列表 leetcode 蓝桥杯

本文链接：https://blog.csdn.net/ZhaoLinke1998/article/details/125136577

版权

c++ 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
        vector<pair<int, int>> array(nums.size());
        count.resize(nums.size(), 0);
        for(int i=0; i<nums.size(); i++){
            array[i] = {i, nums[i]};
        }
        tmp.resize(nums.size());
        sort(array, 0, nums.size()-1);
        return count;
    }
    vector<pair<int, int>> tmp;
    vector<int> count;
    void sort(vector<pair<int, int>>& array, int lo, int hi){
        if(lo >= hi) return;
        int mid = (lo + hi) >> 1;
        sort(array, lo, mid);
        sort(array, mid+1, hi);
        merge(array, lo, mid, hi);
    }
    void merge(vector<pair<int, int>>& array, int lo, int mid, int hi){
        for(int i=lo; i<=hi; i++){
            tmp[i] = array[i];
        }
        int i = lo;
        int j = mid+1;
        for(int p=lo; p<=hi; p++){
            if(i == mid+1){
                array[p] = tmp[j++];
            }
            else if(j == hi+1){
                array[p] = tmp[i++];
                count[array[p].first] += j-mid-1;
            }
            else if(tmp[i].second > tmp[j].second){
                array[p] = tmp[j++];
            }
            else if(tmp[i].second <= tmp[j].second){
                array[p] = tmp[i++];
                count[array[p].first] += j-mid-1;
            }
        }
    }
};