leetcode315. 计算右侧小于当前元素的个数(树状数组解法)
生活随笔
收集整理的這篇文章主要介紹了
leetcode315. 计算右侧小于当前元素的个数(树状数组解法)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
leetcode315. 計(jì)算右側(cè)小于當(dāng)前元素的個(gè)數(shù)(樹狀數(shù)組解法)
題目:給定一個(gè)整數(shù)數(shù)組 nums,按要求返回一個(gè)新數(shù)組 counts。數(shù)組 counts 有該性質(zhì): counts[i] 的值是 nums[i] 右側(cè)小于 nums[i] 的元素的數(shù)量。
樹狀數(shù)組解法
```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]);//當(dāng)前元素的排名helper.update(temp,1);//將當(dāng)前元素的名次更新數(shù)字?jǐn)?shù)組res.add(helper.query(temp-1));//查詢小于當(dāng)前元素的}Collections.reverse(res);return res;}class fenWickTree//樹狀數(shù)組{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;}} }總結(jié)
以上是生活随笔為你收集整理的leetcode315. 计算右侧小于当前元素的个数(树状数组解法)的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 梦到很多毒蛇是什么意思
- 下一篇: 为什么总是梦到自己老公