leetcode 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 个更小的元素.

思路：离散化一下，从后往前更新，每次先查找[0,nums[i]-1]的区间和，找完在更新线段树，把nums[i]更新到
树中。从前往后更新也行。其实是一样的，只是从前往后要找[nums[i]+1,n]的区间和。

class Solution {
    class Idx{
        int dt;
        int idx;
    }
    int[] sum;
    public void pushUp(int rt){
        sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    }
    public void update(int p,int l,int r,int rt){
        if(l==r){
            sum[rt]++;
            return;
        }
        int mid=(l+r)>>1;
        if(p<=mid){
            update(p,l,mid,rt<<1);
        }else{
            update(p,mid+1,r,rt<<1|1);
        }
        pushUp(rt);
    }
    public int query(int L,int R,int l,int r,int rt){
        if(L>R){
            return 0;
        }
        if(L<=l&&r<=R){
            return sum[rt];
        }
        int mid=(l+r)>>1;
        int ans=0;
        if(L<=mid){
            ans+=query(L,R,l,mid,rt<<1);
        }
        if(R>mid){
            ans+=query(L,R,mid+1,r,rt<<1|1);
        }
        return ans;
    }
    public List<Integer> countSmaller(int[] nums) {
        int n=nums.length;
        int[] rank=new int[n];
        Idx[] idxes=new Idx[n];
        for(int i=0;i<n;++i){
            idxes[i]=new Idx();
            idxes[i].dt=nums[i];
            idxes[i].idx=i;
        }
        Arrays.sort(idxes,new Comparator<Idx>() {
            @Override
            public int compare(Idx idx,Idx t1) {
                return idx.dt-t1.dt;
            }
        } );
        int total=0;
        for(int i=0;i<n;++i){
            while (i+1<n&&idxes[i].dt==idxes[i+1].dt) {
                rank[idxes[i].idx]=total;
                ++i;
            }
            rank[idxes[i].idx]=total++;
        }
        for(int i=0;i<n;++i){
            nums[i]=rank[i];
        }
        sum=new int[n<<2];
        List<Integer> ans=new ArrayList<Integer>();
        for(int i=nums.length-1;i>=0;--i){
            ans.add(query(0,nums[i]-1,0,total,1));
            update(nums[i],0,total,1);
        }
        Collections.reverse(ans);
        return ans;
    }
}