Leetcode 528. 按权重随机选择 C++

Leetcode 528. 按权重随机选择

题目

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

例如，给定一个值 [1，9] 的输入列表，当我们从中选择一个数字时，很有可能 10 次中有 9 次应该选择数字 9 作为答案。

测试样例

示例 1：

输入：
["Solution","pickIndex"]
[[[1]],[]]
输出：[null,0]

示例 2：

输入：
["Solution","pickIndex","pickIndex","pickIndex","pickIndex","pickIndex"]
[[[1,3]],[],[],[],[],[]]
输出：[null,0,1,1,1,0]

输入语法说明：

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

提示：

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

题解

我们使用前缀和的方式构造一个权重累加的数组，我们利用这个数组最后一个元素随机生成我们的目标所属权重。使用二分查找方法找到该 权重所属的下标。详细过程见代码

代码

	vector<int> weight;
    Solution(vector<int>& w) {
        weight.push_back(0);
        for(int i=0; i<w.size(); i++)
            weight.push_back(weight.back()+w[i]);
    }
    
    int pickIndex(){
        int w = rand()%weight.back() + 1;
        int l=0,r=weight.size()-1,mid;
        while(l < r){
            mid = (l+r)/2;
            if(weight[mid] == w)    return mid-1;
            else if(weight[mid]<w) l = mid+1;
            else    r = mid;
        }
        return r-1;
    }

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