友情支持

如果您觉得这个笔记对您有所帮助,看在D瓜哥码这么多字的辛苦上,请友情支持一下,D瓜哥感激不尽,😜

支付宝

微信

有些打赏的朋友希望可以加个好友,欢迎关注D 瓜哥的微信公众号,这样就可以通过公众号的回复直接给我发信息。

wx jikerizhi

公众号的微信号是: jikerizhi因为众所周知的原因,有时图片加载不出来。 如果图片加载不出来可以直接通过搜索微信号来查找我的公众号。

398. 随机数索引

给你一个可能含有 重复元素 的整数数组 nums,请你随机输出给定的目标数字 target 的索引。你可以假设给定的数字一定存在于数组中。

实现 Solution 类:

  • Solution(int[] nums) 用数组 nums 初始化对象。

  • int pick(int target)nums 中选出一个满足 nums[i] == target 的随机索引 i。如果存在多个有效的索引,则每个索引的返回概率应当相等。

示例:

输入
["Solution", "pick", "pick", "pick"]
[[[1, 2, 3, 3, 3]], [3], [1], [3]]
输出
[null, 4, 0, 2]

解释
Solution solution = new Solution([1, 2, 3, 3, 3]);
solution.pick(3); // 随机返回索引 2, 3 或者 4 之一。每个索引的返回概率应该相等。
solution.pick(1); // 返回 0 。因为只有 nums[0] 等于 1 。
solution.pick(3); // 随机返回索引 2, 3 或者 4 之一。每个索引的返回概率应该相等。

提示:

  • 1 <= nums.length <= 2 * 104

  • -231 <= nums[i] <= 231 - 1

  • targetnums 中的一个整数

  • 最多调用 pick 函数 104

思路分析

要求重复数字的索引概率相同,那么只需要统计每个数字的索引列表,从列表中随机取即可。

也可以使用蓄水池抽样算法。

  • 一刷

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
/**
 * @author D瓜哥 · https://www.diguage.com
 * @since 2025-08-01 22:46:30
 */
class Solution {
  private Map<Integer, List<Integer>> numToIndex = new HashMap<>();
  private Random random;

  public Solution(int[] nums) {
    for (int i = 0; i < nums.length; i++) {
      numToIndex.computeIfAbsent(nums[i], k -> new ArrayList<>()).add(i);
    }
    random = new Random();
  }

  public int pick(int target) {
    List<Integer> index = numToIndex.get(target);
    return index.get(random.nextInt(index.size()));
  }
}