JAVA程序设计：按权重随机选择（LeetCode：528）

给定一个正整数数组 w ，其中 w[i] 代表位置 i 的权重，请写一个函数 pickIndex ，它可以随机地获取位置 i，选取位置 i 的概率与 w[i] 成正比。

说明:

1 <= w.length <= 10000
1 <= w[i] <= 10^5
pickIndex 将被调用不超过 10000 次
示例1:

输入: 
["Solution","pickIndex"]
[[[1]],[]]
输出: [null,0]
示例2:

输入: 
["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
[[[1,3]],[],[],[],[],[]]
输出: [null,0,1,1,1,0]
输入语法说明：

输入是两个列表：调用成员函数名和调用的参数。Solution 的构造函数有一个参数，即数组 w。pickIndex 没有参数。输入参数是一个列表，即使参数为空，也会输入一个 [] 空列表。

思路：求一波前缀和，在区间[1,sum[len-1]]里随机一个数，二分看他在哪个位置上即可。

class Solution {

	private int len;
	private int[] sum;
	
    public Solution(int[] w) {
    	
    	len=w.length;
    	sum=new int[len];
        for(int i=0;i<len;i++)
        	sum[i]=w[i]+(i==0?0:sum[i-1]);
        
    }
    
    public int pickIndex() {
        int pos=(int)(Math.random()*sum[len-1])+1;
        int l=0,r=len-1;	
        while(l<r) {
        	int mid=(l+r)/2;
        	if(pos<=sum[mid])
        		r=mid;
        	else
        		l=mid+1;
        }
        return l;
    }
}