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