398. 随机数索引

题目

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

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

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

代码

class Solution {
    int[] res;
    int[] nums;
    public Solution(int[] nums) {
       this.nums=nums;
    }

    public int pick(int target) {
        Random r = new Random();
        int n=0,index=0;

        // 前 k 个元素先默认选上
        for (int i = 0; i <nums.length ; i++) {
            if (nums[i]==target) {//每找到一个目标
                n++;//我们以1/n的概率留下该数据
                if (r.nextInt(n)==0) 
                    index=i;
            }
        }

        return index;
    }
}

要点

• 室友水池抽样法，对于每个num[i]=target，都有1/n的概率留下该索引
• 蓄水池抽样（Random Reservior Sampling) 保存整个nums数组。O(N)额外空间