给定一个整数数组 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;
}
}