528. 按权重随机选择

发表于 更新于 分类于
这是文章开头，显示在主页面，详情请点击此处。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution:

    def __init__(self, w: List[int]):
        self.pre = list(accumulate(w))
        self.sum_ = sum(w)
        print(self.sum_)
    def pickIndex(self) -> int:
        x = random.randint(1,self.sum_)
        return bisect.bisect_left(self.pre,x)

    


# Your Solution object will be instantiated and called as such:
# obj = Solution(w)
# param_1 = obj.pickIndex()