8-30每日一题：LeetCode 528. 按权重随机选择

最新推荐文章于 2025-05-23 02:00:00 发布

原创最新推荐文章于 2025-05-23 02:00:00 发布 · 151 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

每日一题专栏收录该内容

144 篇文章

订阅专栏

本文介绍了解决LeetCode题目528的方法，通过预处理前缀和，将给定权重数组映射到数轴，并利用随机数生成随机索引，借助二分查找确定随机点落在哪个区间。讲解了Solution类的实现细节和关键函数pickIndex()的工作原理。

LeetCode 528. 按权重随机选择

思路

预处理前缀和，将数组映射到数轴上，产生数组长度的随机数，判断落点位置，二分查找属于哪一个区间

class Solution {
public:
    vector<int> s;

    Solution(vector<int>& w) {
        s = w;
        for (int i = 1; i < w.size(); i ++ ) {
            s[i] += s[i - 1];
        }
    }
    
    int pickIndex() {
        int x = rand() % s.back() + 1;
        return lower_bound(s.begin(), s.end(), x) - s.begin();
    }
};

/**
 * Your Solution object will be instantiated and called as such:
 * Solution* obj = new Solution(w);
 * int param_1 = obj->pickIndex();
 */