leetcode315. 计算右侧小于当前元素的个数(树状数组解法)
题目:给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
树状数组解法
```java
class Solution {
public List<Integer> countSmaller(int[] nums) {
ArrayList<Integer> res=new ArrayList<>();
int n=nums.length;
if(n==0) return res;
//利用二叉搜索树
Set<Integer> set=new TreeSet<>();
for(int c:nums) set.add(c);
//生成排名表
int level=1;
HashMap<Integer,Integer> map=new HashMap<>();
for(int c:set)
{
map.put(c,level);
level++;
}
fenWickTree helper=new fenWickTree(set.size()+1);
for(int i=n-1;i>=0;i--)
{
int temp=map.get(nums[i]);//当前元素的排名
helper.update(temp,1);//将当前元素的名次更新数字数组
res.add(helper.query(temp-1));//查询小于当前元素的
}
Collections.reverse(res);
return res;
}
class fenWickTree//树状数组
{
int[] sum;
int len;
public fenWickTree(int n)
{
len=n;
sum=new int[n+1];
}
public int lowBits(int x)
{
return x&(-x);
}
public void update(int i,int num)
{
while (i<=len)
{
sum[i]+=num;
i+=lowBits(i);
}
}
public int query(int i)
{
int res=0;
while (i>0)
{
res+=sum[i];
i-=lowBits(i);
}
return res;
}
}
}