LeetCode 528. 按权重随机选择（前缀和+二分查找）

文章目录

1. 题目
2. 解题

1. 题目

给定一个正整数数组 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。示例 2：
输入：
["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
[[[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 次

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/random-pick-with-weight
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：
LeetCode 497. 非重叠矩形中的随机点（前缀和+二分查找）

计算前缀和权重
随机权值，二分查找，找到权值落在的区间点

class Solution {vector<int> w_presum;int total;
public:Solution(vector<int>& w) {w_presum = w;for(int i = 1; i < w.size(); ++i)w_presum[i] += w_presum[i-1];total = w_presum.back();}int pickIndex() {int w = rand()%total + 1;int L = 0, R = w_presum.size()-1, mid;while(L <= R){mid = L+((R-L)>>1);if(w_presum[mid] < w)L = mid+1;else{if(mid==0 || w_presum[mid-1] < w)return mid;elseR = mid-1;}}return -1;}
};

88 ms 39.3 MB C++

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！

LeetCode 528. 按权重随机选择（前缀和+二分查找）相关推荐

Java实现 LeetCode 528 按权重随机选择（TreeMap）
528. 按权重随机选择给定一个正整数数组 w ,其中 w[i] 代表位置 i 的权重,请写一个函数 pickIndex ,它可以随机地获取位置 i,选取位置 i 的概率与 w[i] 成正比. 说明 ...
[528]. 按权重随机选择
[528]. 按权重随机选择题目算法设计:加权随机取样题目传送门:528. 按权重随机选择输入: ["Solution","pickIndex"] [ ...
528. 按权重随机选择
528. 按权重随机选择
LeetCode 497. 非重叠矩形中的随机点（前缀和+二分查找）
文章目录 1. 题目 2. 解题 1. 题目给定一个非重叠轴对齐矩形的列表 rects,写一个函数 pick 随机均匀地选取矩形覆盖的空间中的整数点. 提示: 整数点是具有整数坐标的点. 矩形周边上 ...
随机数生成生成器和力扣按权重随机选择 528
随机数生成生成器和力扣按权重随机选择 528 在linux下内核有个专门内核模块进行生成随机数 /dev/random 不是很精准 /dev/urandom 更精准原理是内核的一个模块,专门产生的 ...
挑战程序设计竞赛部分题目选择排序、二分查找找出下标、深度优先算法、求三角形max周长、蚂蚁算法、操作excel
首先提供帮助类可能这个类还需要更新 .求最大值.求最小值.选择排序.二分查找找出下标.二分查找排好序列里面是否有那个值 public class Util {//求最大值public static ...
LeetCode 729. 我的日程安排表 I（set 二分查找）
文章目录 1. 题目 2. 解题 2.1 set 二分查找 2.2 差分思想 1. 题目实现一个 MyCalendar 类来存放你的日程安排.如果要添加的时间内没有其他安排,则可以存储这个新的日程安 ...
209 长度最小的子数组（前缀和+二分查找、滑动窗口）
1. 问题描述: 给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组,并返回其长度.如果不存在符合条件的子数组,返回 0. 示例: 输入:s = ...
[Golang]力扣Leetcode - 852. 山脉数组的峰顶索引（二分查找）
[Golang]力扣Leetcode - 852. 山脉数组的峰顶索引(二分查找) 题目:符合下列属性的数组 arr 称为山脉数组 : arr.length >= 3 存在 i(0 < ...

LeetCode 528. 按权重随机选择（前缀和+二分查找）

文章目录

1. 题目

2. 解题

LeetCode 528. 按权重随机选择（前缀和+二分查找）相关推荐

最新文章

热门文章