Random Pick Index Solutions in C++

// Source : https://leetcode.com/problems/random-pick-index/
// Author : Hao Chen
// Date   : 2016-11-04



class Solution {
private:
    vector<int> nums;    
public:
    Solution(vector<int> nums) {
        srand(time(0));
        this->nums = nums;
    }
    
    int pick(int target) {
	// we just randomly pick a number from the array,
        // if the number is target just return the index.
        // otherwise, keep picking the number randomly.
        while(true) {
            int idx = rand() % nums.size();
            if ( target == nums[idx] ) {
                return idx;
            }
        }
    }
};

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

/// Source : https://leetcode.com/problems/random-pick-index/description/
/// Author : liuyubobobo
/// Time   : 2018-07-08

#include <iostream>
#include <vector>

using namespace std;


/// Reservor Sampling
/// The detail of Resevoir Sampling can be seen here:
/// https://www.geeksforgeeks.org/reservoir-sampling/
///
/// Attention: Don't use srand in the random problem in Leetcode
/// Since it may lead to WA
/// I think the test code use its own random seed to verify your submitted code
///
/// Time Complexity: init: O(n)
///                  pick: O(n)
/// Space Complexity: O(1)
class Solution {

private:
    vector<int> nums;

public:
    Solution(vector<int> nums) {

        for(int num: nums)
            this->nums.push_back(num);
    }

    int pick(int target) {

        int res = -1;
        int cnt = 0;
        for(int i = 0 ; i < nums.size() ; i ++)
            if(nums[i] == target){
                cnt ++;
                int rnd = rand() % cnt;
                if(rnd == 0)
                    res = i;
            }
        return res;
    }
};


int main() {

    return 0;
}