Leetcode 528. 按权重随机选择（按权重映射到数轴上，随机数抽样选择）

最新推荐文章于 2022-04-01 18:57:16 发布

wwxy261

最新推荐文章于 2022-04-01 18:57:16 发布

阅读量157

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/119988337

版权

前缀和二分查找随机选择权重概率分布

关键词由CSDN通过智能技术生成

算法专栏收录该内容

3633 篇文章 120 订阅

订阅专栏

前缀和加二分查找, 即将整个区间映射到0到1的数轴上，然后生成一个0到1的随机数，然后判断该随机数处在哪个区间上。

class Solution:

    def __init__(self, w: List[int]):
        probability = [0] * len(w)
        _sum = sum(w)
        cur = 0
        for i in range(len(w)):
            cur += w[i]
            probability[i] = cur / _sum
        self.probability = probability


    def pickIndex(self) -> int:
        return bisect.bisect_right(self.probability, random.random())