数据结构与算法之深入解析“按权重随机选择”的求解思路与算法示例

Posted 2022-03-23 Serendipity·y

一、题目要求

• 给你一个下标从 0 开始的正整数数组 w ，其中 w[i] 代表第 i 个下标的权重，请你实现一个函数 pickIndex，它可以随机地从范围 [0, w.length - 1] 内（含 0 和 w.length - 1）选出并返回一个下标，选取下标 i 的概率 为 w[i] / sum(w) 。
• 例如，对于 w = [1, 3]，挑选下标 0 的概率为 1 / (1 + 3) = 0.25 （即 25%），而选取下标 1 的概率为 3 / (1 + 3) = 0.75（即 75%）。
• 示例 1：

输入：
["Solution","pickIndex"]
[[[1]],