二分查找26（Leetcode528按权重随机选择）

最新推荐文章于 2024-10-11 12:01:25 发布

从月亮走向月亮7

最新推荐文章于 2024-10-11 12:01:25 发布

阅读量362

点赞数 4

文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/stacey777/article/details/134964442

版权

代码：

看不懂题看了题解才懂题是什么意思..

二分查找完结撒花！🎉

class Solution {
    int[] pre;
    int total;

    public Solution(int[] w) {
        int n = w.length;
        total = Arrays.stream(w).sum();
        pre = new int[n];
        pre[0] = w[0];
        for(int i=1;i<n;i++){
            pre[i] = pre[i-1] + w[i];
        }
    }
    
    public int pickIndex() {
        int x = (int)(Math.random()*total) +1;
        return binarySearch(x);
        
    }
    public int binarySearch(int x){
        int l=0,r=pre.length-1;
        while(l<r){
            int m = l + (r-l)/2;
            if(pre[m]<x){
                l = m+1;
            }else{
                r = m;
            }
        }
        return l;
    }

}