【leetcode 蓄水池抽样 C++】398. Random Pick Index

398. Random Pick Index

class Solution {
public:
    vector<int> V;
    Solution(vector<int>& nums) {
        V = nums;
    }
    
    int pick(int target) {
        int ans = 0;
        int cnt = 0;
        for(int ii = 0; ii < V.size(); ii++) {
            if(V[ii] == target && rand() % ++cnt == 0) ans = ii;
        }
        return ans;
    }
};

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