398. 随机数索引 :「哈希表预处理」&「蓄水池抽样」-CFANZ编程社区

题目描述

这是 LeetCode 上的 398. 随机数索引 ，难度为中等。

Tag : 「哈希表」、「模拟」、「随机化」、「蓄水池抽样」

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

注意：

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

示例:

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);

提示：

target 确保存在于nums 中
最多调用次的pick

哈希表预处理（定长数据流）

切换英文补全一下数据范围：

target 确保存在于nums 中
最多调用次的pick

为了方便，我们令 nums 的长度为，利用的数据范围为，且完整的数组为初始化时已给出，我们可以通过使用「哈希表 + 预处理」的方式进行求解。

具体的，在构造函数传入 nums 时，遍历 nums 并存储每个对应的下标集合，即使用哈希表以为键，下标集合 List 作为值进行存储。

在 pick 操作时，通过的复杂度取出所有的集合下标，再随机一个下标进行返回。

代码：

class Solution {
    Random random = new Random();
    Map<Integer, List<Integer>> map = new HashMap<>();
    public Solution(int[] nums) {
        int n = nums.length;
        for (int i = 0; i < n; i++) {
            List<Integer> list = map.getOrDefault(nums[i], new ArrayList<>());
            list.add(i);
            map.put(nums[i], list);
        }
    }
    public int pick(int target) {
        List<Integer> list = map.get(target);
        return list.get(random.nextInt(list.size()));
    }
}

时间复杂度：初始化的复杂度为；pick 操作的复杂度为
空间复杂度：

蓄水池抽样（不定长数据流）

若 nums 并不是在初始化时完全给出，而是持续以「流」的形式给出，且数据流的很长，不便进行预处理的话，我们只能使用「蓄水池抽样」的方式求解。

不了解「蓄水池抽样」的同学可以看前置 ???? : 多语言入门「蓄水池抽样」知识点

具体的，我们在每次 pick 时对流进行遍历，由于数据流很大，我们不能在遍历过程中使用诸如数组的容器存储所有满足条件的下标，只能对于每个执行「是否要将作为最新答案候选」的操作。

假设共有个下标满足，我们需要做到以概率返回任一坐标。

我们规定当遇到第个满足的下标时，执行一次的随机操作，当随机结果为时（发生概率为），我们将该坐标作为最新的答案候选。

当对每一个的下标都进行上述操作后，容易证明每一位下标返回的概率均为。

假设最后返回的是第个满足条件的下标，发生概率为 = 第个下标被候选的概率后面到个下标不被候选的概率 = = 。

代码：

class Solution {
    Random random = new Random();
    int[] nums;
    public Solution(int[] _nums) {
        nums = _nums;
    }
    public int pick(int target) {
        int n = nums.length, ans = 0;
        for (int i = 0, cnt = 0; i < n; i++) {
            if (nums[i] == target) {
                cnt++;
                if (random.nextInt(cnt) == 0) ans = i;
            }
        }
        return ans;
    }
}