*Leetcode 528. 按权重随机选择

Z-Pilgrim

于 2023-04-02 14:32:10 发布

阅读量120

点赞数

文章标签： leetcode c++ 算法

本文链接：https://blog.youkuaiyun.com/u011026968/article/details/129910625

版权

该文章描述了一个在C++中实现的Solution类，用于根据给定权重向量w生成随机索引。类初始化时计算了前缀和pre_sum，并使用srand和rand函数生成随机数。pickIndex方法利用二分查找找到随机数对应的索引位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

等于接口题。。。随机数生成接口

力扣

class Solution {
public:
    Solution(vector<int>& w){
        if(w.size()) {
            int cur = w[0];
            pre_sum.push_back(cur);
            for(int i = 1; i < w.size(); i++) {
                cur += w[i];
                pre_sum.push_back(cur);
            }
            srand((int)time(0));
        }
    }
    
    int pickIndex() {
        // int randv = (dis(mt_rand)+1) % pre_sum.back();
        int randv = rand()%pre_sum.back() + 1;
        return lower_bound(pre_sum.begin(), pre_sum.end(), randv) - pre_sum.begin();
    }
    vector<int> pre_sum;
};

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