剑指 Offer II 071. 按权重生成随机数

题意：

题意概述：给一个数x，找出该数在数组w中的下标，但是数组中的数是要按权重来算的，权重大的他出现的概率大，意味着它的下标被获取的概率大。
解题思路：用total来计算数组w的累加和，然后把1-total分成长度分别为w[0],w[1]…w[n]的各段，例如total=10,w=[1,3,5,1]，分成[1],[2,3,4],[5,6,7,8,9],[10]对应的长度分别为1,3,5,1，刚好符合它们各自所占的权重。用sum[i]来记录前缀和，对于一个数x，要满足：sum[i - 1] + 1≤x≤sum[i]，则i就是它要找的下标。
这道题可以用二分来查找，因为sum[i]是递增的，所以只需要找出满足x≤sum[i]的最小的i即可。
这道题很妙的就是，它用总的和来划分每一段对应的长度，每一段就对应着下标i出现的概率，不错!

class Solution {
    public long[] sum;
    public long total;
    public Solution(int[] w) {
        sum = new long[w.length];
        sum[0] = w[0];
        total = w[0];
        for(int i = 1; i < w.length; i++) {
            sum[i] = sum [i - 1] + w[i];
            total += w[i];
        }
    }   
    public int pickIndex() {
        int x = (int)(Math.random() * total + 1);
        int left = 0, right = sum.length - 1;
        while(left < right) {
            int mid = left + right >> 1;
            if(x > sum[mid]) {
                left = mid + 1;
            } else {
                right = mid;
            }
        }
        return left;
    }
}