leetcode-398-随机数索引

class Solution {
    
    int[] nums;
    Random random;

    public Solution(int[] nums) {
        this.nums = nums;
        random = new Random();
    }
    
    public int pick(int target) {
        int result = -1;
        int count = 1;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != target) {
                continue;
            }
            // 随机生成 [0, count) 之间的整数
            // 整数等于 0 的概率为 1/count
            if (random.nextInt(count) == 0) {
                result = i;
            }
            count++;
        }
        return result;
    }
}

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