Random Pick Index

random_pick_index.cpp

class Solution {
private:
    vector<int> nums;
public:
    Solution(vector<int> nums) {
        this->nums = nums;
        srand(time(NULL));
    }
    
    int pick(int target) {
        vector<int> indices;
        for (int i = 0; i < nums.size(); ++i) {
            if (nums[i] == target)
                indices.push_back(i);
        }
        
        return indices[rand() % indices.size()];
    }
};