【Lintcode】249. Count of Smaller Number before itself

题目地址：

https://www.lintcode.com/problem/count-of-smaller-number-before-itself/description

给定一个数组$A$，其范围是$0\sim 10000$。要求返回一个数组$B$使得$B[i]$是$A[i]$之前比$A[i]$小的数的个数。

可以用树状数组做。先求出$A$中数字的最大值，比如说是$k$，想象一个数组$C$，其长度为$k+1$，并且如果$x\in A$则$C[x]=1$，否则$C[x]=0$。这样要求$A[i]$前面比$A[i]$小的数的个数（记为$f(A[i])$）等价于求$C$数组的前$A[i]$个数的和（$C$要随着对$A$的遍历逐次更新）。比如，如果$A=[7,8,2,1,3]$，则在遍历到$8$的时候，$C=[0,0,0,0,0,0,0,1,1]$（也就是$C[7]$和$C[8]$更新为$1$），$f(8)=C[0]+C[1]+...+C[7]$也就是$C$数组前$8$个数的和。而求前缀和可以用树状数组实现。

由于树状数组的数组下标是从$1$开始计的，$0$的位置被废弃了，所以代码实现方面要尤其注意下标。具体代码方面要这样做：
1、先找到$A$的最大值，设为$m$，想象一个长度为$m+1$的数组$C$，这样$C[0,...,m]$都有意义；
2、开一个树状数组用来维护数组$C$的前缀和，那么树状数组本身的长度应该是$C$的长度加$1$，也就是$m+2$。
3、开始遍历$A$，当遍历到$A[i]$的时候，就在$C[A[i]]$上加上$1$，树状数组也要相应更新，然后顺便算一下$C$的前$A[i]$个数的和（这里前$j$个数的$j$是从$1$开始计数的），这个和就是$A[i]$之前比$A[i]$小的数的个数。

代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    
    class FenwickTree {
        private int[] tree;
    
        public FenwickTree(int n) {
            tree = new int[n + 1];
        }
        
        // 在原数组的第i个位置增加x（i从1开始计数），对应的树状数组的更新
        public void add(int i, int x) {
            while (i < tree.length) {
                tree[i] += x;
                i += lowbit(i);
            }
        }
        
        // 计算原数组的前i个数字的和（i也是从1开始计数，前0个数的和记为0）
        public int sum(int i) {
            int res = 0;
            while (i > 0) {
                res += tree[i];
                i -= lowbit(i);
            }
            
            return res;
        }
        
        private int lowbit(int x) {
            return x & -x;
        }
    }
    
    /**
     * @param A: an integer array
     * @return: A list of integers includes the index of the first number and the index of the last number
     */
    public List<Integer> countOfSmallerNumberII(int[] A) {
        // write your code here
        List<Integer> res = new ArrayList<>();
        
        // 求一下最大值
        int max = 0;
        for (int i : A) {
            max = Math.max(max, i);
        }
        
        // 开一个树状数组，维护一个长度为max + 1的数组C
        FenwickTree fenwickTree = new FenwickTree(max + 1);
        for (int i = 0; i < A.length; i++) {
        	// 在C的第A[i] + 1个位置上的数加上1
            fenwickTree.add(A[i] + 1, 1);
            // 求C的前A[i]个数的和
            res.add(fenwickTree.sum(A[i]));
        }
        
        return res;
    }
}

时间复杂度$O(n\log max)$，空间$O(max)$。

注解：
由于树状数组的tree下标是从$1$开始的，所以下标极其容易搞错。我们只需要这样做就就能尽可能避免在下标上犯错：
1、树状数组的模板记牢，始终用同一个模板，模板中每个变量的含义搞清楚。比如说，构造器传的参数n，就是指要维护的数组的长度；然后add和query中的参数i，都是指原数组的第i个数，i从1开始取。add就是在原数组的第i个数上加上某个数，query就是求原数组的前i个数之和；
2、在模板之外的代码里，进行操作的时候要牢记，我们操作的是被维护数组的信息，树状数组已经被封装起来了，不要把外面的逻辑带到树状数组里去。比如fenwickTree.add(A[i] + 1, 1);，这是在让被维护数组的第A[i] + 1个数加上了1。我们不需要管树状数组是怎么做的，只需要搞清楚在被维护数组上做了什么修改即可。