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

最新推荐文章于 2023-12-25 11:23:38 发布

qbnmm

最新推荐文章于 2023-12-25 11:23:38 发布

阅读量103

点赞数

分类专栏：刷题笔记文章标签：算法 leetcode

本文链接：https://blog.csdn.net/qq_33523932/article/details/125677556

版权

刷题笔记专栏收录该内容

118 篇文章 1 订阅

订阅专栏

本文介绍如何使用Python解决力扣LeetCode中的随机选取下标问题，通过计算权重前缀和并结合二分查找，确保选取概率与权重成正比。实例展示了如何创建Solution类并实现pickIndex方法来模拟概率选择过程。

摘要由CSDN通过智能技术生成

题目链接

力扣

题目描述

给定一个正整数数组 w ，其中 w[i] 代表下标 i 的权重（下标从 0 开始），请写一个函数 pickIndex ，它可以随机地获取下标 i，选取下标 i 的概率与 w[i] 成正比。

例如，对于 w = [1, 3]，挑选下标 0 的概率为 1 / (1 + 3) = 0.25 （即，25%），而选取下标 1 的概率为 3 / (1 + 3) = 0.75（即，75%）。

也就是说，选取下标 i 的概率为 w[i] / sum(w) 。

示例 1：

输入：
inputs = ["Solution","pickIndex"]
inputs = [[[1]],[]]
输出：
[null,0]
解释：
Solution solution = new Solution([1]);
solution.pickIndex(); // 返回 0，因为数组中只有一个元素，所以唯一的选择是返回下标 0。
示例 2：

输入：
inputs = ["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
inputs = [[[1,3]],[],[],[],[],[]]
输出：
[null,1,1,1,1,0]
解释：
Solution solution = new Solution([1, 3]);
solution.pickIndex(); // 返回 1，返回下标 1，返回该下标概率为 3/4 。
solution.pickIndex(); // 返回 1
solution.pickIndex(); // 返回 1
solution.pickIndex(); // 返回 1
solution.pickIndex(); // 返回 0，返回下标 0，返回该下标概率为 1/4 。

由于这是一个随机问题，允许多个答案，因此下列输出都可以被认为是正确的:
[null,1,1,1,1,0]
[null,1,1,1,1,1]
[null,1,1,1,0,0]
[null,1,1,1,0,1]
[null,1,0,1,0,0]
......
诸若此类。

提示：

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

解题思路

思路：随机+二分查找
将每个元素的权重累加前面的权重，变成前缀和，然后随机生成1到sum(w)的整数，二分查找在前缀和数组的下标
坐标就是结果

代码

Python

class Solution:

    def __init__(self, w: list[int]):
        n = len(w)
        self.w = [0] * n
        self.w[0] = w[0]
        for i in range(1, n):
            self.w[i] = self.w[i - 1] + w[i]

    def pickIndex(self) -> int:
        rand = random.randint(1, self.w[-1])
        return bisect.bisect_left(self.w, rand)