LeetCode 528 按权重随机选择[前缀和 二分法] HERODING的LeetCode之路

Posted 2021-09-15 HERODING23

解题思路：
题目有些含糊不清，其实就是输入一组权重，代表当前位置输出的权重，求输出概率，解决这道题的关键在于前缀和，把每个位置的权重像标记刻度一样记录下来，落到哪个区间就输出哪个下标，落到区间使用二分法查找，代码如下：

class Solution {
    vector<int> w;
public:
    Solution(vector<int>& W) {
        w.push_back(W[0]);
        for(int i = 1; i < W.size(); i ++) {
            w.push_back(w.back() + W[i]);
        }
    }
    
    int pickIndex() {
        if(w.size() == 1) {
            return 0;
        }
        int index = rand() % w.back() + 1;
        int position = lower_bound(w.begin(), w.end(), index) - w.begin();
        return position;
    }
};