LeetCode 按权重随机选择（二分法）

最新推荐文章于 2022-06-25 10:16:52 发布

hestyle

最新推荐文章于 2022-06-25 10:16:52 发布

阅读量1.3k

点赞数

分类专栏： LeetCode 二分法

本文链接：https://blog.csdn.net/qq_41855420/article/details/89109111

版权

这篇博客介绍了如何解决一个LeetCode题目，即根据权重随机选择数组中的位置。文章指出，通过将权重映射到线段位置，并使用二分查找优化查找过程，可以提高效率，将原本线性的查找时间降低到对数级别。

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

给定一个正整数数组 w ，其中 w[i] 代表位置 i 的权重，请写一个函数 pickIndex ，它可以随机地获取位置 i，选取位置 i 的概率与 w[i] 成正比。
说明:

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

示例1:

输入: 
["Solution","pickIndex"]
[[[1]],[]]
输出: [null,0]
示例2:
输入: 
["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
[[[1,3]],[],[],[],[],[]]
输出: [null,0,1,1,1,0]

输入语法说明：

输入是两个列表：调用成员函数名和调用的参数。Solution 的构造函数有一个参数，即数组 w。
pickIndex 没有参数。输入参数是一个列表，即使参数为空，也会输入一个 [] 空列表。

思路分析： 将w[i]映射到线段的位置，然后穷举线段上的位置即可。
在这里插入图片描述

class Solution {
   
    int sum = 0, numsSize = 0;
    vector<int> numsEnd;
public:
    Solution(vector<int>& w) {
   
        sum = 0;
        numsSize