Leetcode528按权重随机选择解题思路

前言

按权重随机选择在开发中是一个相对于其他leetcode题来说比较实用，且在日常开发中会见到和使用的算法，还是有现实场景应用的。所以今天leetcode每日刷题AC后，觉得还是有必要做个笔记总结下。

Leetcode 528.按权重随机选择

给定一个正整数数组 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：

输入：

[“Solution”,“pickIndex”]

[[[1]],[]]

输出：

[null,0]

解释：

Solution solution = new Solution([1]);

solution.pickIndex(); // 返回 0，因为数组中只有一个元素，所以唯一的选择是返回下标 0。

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

---

读题：

1. w[i] 代表下标 i 的权重。
2. 选下标 i 的概率为 w[i] / sum(w)，即此权重值除以整个数组的累加和。

要求：写个函数pickIndex，使能够随机获得下标 i ，选取下标 i 的概率与 w[i] 成正比。

场景例子

比如：假如有10个游戏道具，有屠龙宝刀，也有烧火棍等等。需要随机抽出一个发给玩家。由于道具的稀缺度，价值等不同，所以抽取的概率肯定不希望是是平均的。也就是高价值高权重获得的概率低，低价值低权重获得的概率高。

要实现这种效果，就需要给每个道具指定一个权重。比如烧火棍比较垃圾给个权重80，屠龙宝刀稀有给个权重20。即80%的概率会抽到烧火棍，这个就是基于权重的随机算法。

本题意思相对比较简单，与上边的场景例子期望相反。期望对数组的元素随机采样，元素的权重比越高，出现的概率越高。

解题

以 w = [1, 2, 4, 3]为例：

sum 是10。相当于有10个矩形格子。按照1、2、4、3来划分4个区间。

我们在这10之内随机取一个数。

• 例如1，落在了第1个区间，对应的index就是0。
• 例如2，落在了第2个区间，对应的index就是1。
• 例如3，落在了第2个区间，对应的index就是1。
• 例如4，落在了第3个区间，对应的index就是2。
• 例如5，落在了第3个区间，对应的index就是2。
• 例如6，落在了第3个区间，对应的index就是2。
• 例如7，落在了第3个区间，对应的index就