398. 随机数索引（中等 水塘抽样 随机 设计 数学）

398. 随机数索引

给定一个可能含有重复元素的整数数组，要求随机输出给定的数字的索引。 您可以假设给定的数字一定存在于数组中。

注意：
数组大小可能非常大。 使用太多额外空间的解决方案将不会通过测试。

示例:

int[] nums = new int[] {1,2,3,3,3};
Solution solution = new Solution(nums);

// pick(3) 应该返回索引 2,3 或者 4。每个索引的返回概率应该相等。
solution.pick(3);

// pick(1) 应该返回 0。因为只有nums[0]等于1。
solution.pick(1);

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/random-pick-index
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

水塘抽样问题。
我们假设一共有n个重复元素，在遍历的过程中遍历到第n - 1个的时候，令输出该索引的概率为1 / (n - 1)，（因为不知道是否还有，所以相当于假设只有n - 1个数），当找到第n个的时候，令输出该索引的概率为1 / n，那么前n - 1个数的概率就是(n - 1) / n，根据概率知识，输出第n - 1个数的概率应该是 (n - 1) / n * 1 / (n - 1) = 1 / n。
那么需要考虑的就是如何实现这个思路，这里使用的方法是从n个数（0 ~ n - 1）中选出一个随机数，因为从n个数中选取任何一个数的概率是一样的，所以我们直接在该随机数是0的时候就输出当前元素，因为遍历到第一个数的时候该随机数一定是0，所以不用考虑是否会导致随机不到0的情况。

题解（Java）

class Solution {
    private int[] nums;

    public Solution(int[] nums) {
        this.nums = nums;
    }
    
    public int pick(int target) {
        int n = 0, index = 0;
        Random r = new Random();
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == target) {
                n++;
                if (r.nextInt(n) == 0) index = i;
            }
        }
        return index;
    }
}

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