No.85 - LeetCode315 - 求数组中每个数其右小于它的数量

最新推荐文章于 2024-09-08 14:56:34 发布

原创最新推荐文章于 2024-09-08 14:56:34 发布 · 204 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种基于归并排序的算法，用于计算数组中逆序对的数量。通过自定义节点结构和比较函数，实现了对数组元素位置和值的同时跟踪，有效地解决了逆序对计数问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

归并排序思想：

写了半个小时，没想到一遍过了。

class Solution {
public:
    struct Node{
        int loc;
        int val;
        int cnt;
        Node(){}
        Node(int a,int b,int c):loc(a),val(b),cnt(c){}
    };
    static bool cmp(const Node& x,const Node& y){
        if(x.loc < y.loc) return true;
        return false;
    }
    static void Ncopy(Node& x, Node& y){
        x.loc = y.loc;x.val = y.val;x.cnt = y.cnt;
    }
    void merge_sort(vector<Node>& v,int L,int R){
        if(L+1 >= R) return ;
        int mid = (L+R)/2;
        merge_sort(v,L,mid);
        merge_sort(v,mid,R);
        vector<Node> nv(R-L);
        int i,j;
        i = j = 0;
        while(mid+i<R && L+j<mid){
            if(v[mid+i].val < v[L+j].val){
                Ncopy(nv[i+j],v[mid+i]);
                i++;
            }else{
                Ncopy(nv[i+j],v[L+j]);
                nv[i+j].cnt += i;
                j++;
            }
        }
        for(;mid+i<R;i++) Ncopy(nv[i+j],v[mid+i]);
        for(;L+j<mid;j++){
            Ncopy(nv[i+j],v[L+j]);
            nv[i+j].cnt += i;
        }
        for(int k=0;k<R-L;k++) Ncopy(v[L+k],nv[k]);
    }
    vector<int> countSmaller(vector<int>& nums) {
        int N = nums.size();
        vector<Node> v(N);
        for(int i=0;i<N;i++){
            v[i].loc = i;
            v[i].val = nums[i];
            v[i].cnt = 0;
        }
        merge_sort(v,0,N);
        sort(v.begin(),v.end(),cmp);
        vector<int> ans;
        for(int i=0;i<N;i++) ans.push_back(v[i].cnt);
        return ans;
    }
};