前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 315. Count of Smaller Numbers After Self(线段树,树状数组)

LeetCode 315. Count of Smaller Numbers After Self(线段树,树状数组)

作者头像
ShenduCC
发布2020-03-17 14:07:45
5070
发布2020-03-17 14:07:45
举报
文章被收录于专栏:算法修养算法修养

题目

题意:找到数组里每个元素的右边有多少个元素小于当前元素

题解:单点更新,区间查询。线段树或者树状数组都可以。注意要离散化

代码语言:javascript
复制
class Solution {
public:
    int c[100005];
    int n;
    map<int,int> m;
    vector<int> countSmaller(vector<int>& nums) {
        
        vector<int> a = nums;
        sort(a.begin(),a.end());
        for(int i=0;i<a.size();i++)
        {
            m[a[i]]=i+1;
        }
        n=nums.size();
        vector<int> ans;
        
        for(int i=n-1;i>=0;i--)
        {
            ans.push_back(sum(m[nums[i]]-1));
            update(m[nums[i]]);
        }
        
        reverse(ans.begin(),ans.end());
        return ans;
            
    }
    
    int lowbit(int x)
    {
        return x&(-x);
    }
    
    void update(int pos)
    {
        while(pos<=n)
        {
            c[pos]++;
            pos+=lowbit(pos);
        }
    }
    
    int sum(int pos)
    {
        int ans=0;
        while(pos>0)
        {
            ans+=c[pos];
            pos-=lowbit(pos);
        }
        return ans;
    }
};
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-03-13 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档