记一次Leetcode随机数的题

题目比较新，随机数，以前还没遇到过。官方的题解，用stl用得出神入化，故记录下。

528题，题目描述：

给定一个正整数数组 w ，其中 w[i] 代表下标 i 的权重（下标从 0 开始），请写一个函数 pickIndex ，它可以随机地获取下标 i，选取下标 i 的概率与 w[i] 成正比。

例如，对于 w = [1, 3]，挑选下标 0 的概率为 1 / (1 + 3) = 0.25 （即，25%），而选取下标 1 的概率为 3 / (1 + 3) = 0.75（即，75%）。

也就是说，选取下标 i 的概率为 w[i] / sum(w) 。

示例 1：

输入：
["Solution","pickIndex"]
[[[1]],[]]
输出：
[null,0]
解释：
Solution solution = new Solution([1]);
solution.pickIndex(); // 返回 0，因为数组中只有一个元素，所以唯一的选择是返回下标 0。

代码如下：

class Solution {
public:
    mt19937 gen;
    uniform_int_distribution<int> dis;
    vector<int> pre;
    
    Solution(vector<int>& w):gen(random_device{}()), dis(1, accumulate(w.begin(), w.end(), 0)) {
        partial_sum(w.begin(), w.end(), back_inserter(pre));
    }
    
    int pickIndex() {
        int v = dis(gen);
        return lower_bound(pre.begin(), pre.end(), v) - pre.begin();
    }
};

关于上述用到的stl说明：