友情支持
如果您觉得这个笔记对您有所帮助,看在D瓜哥码这么多字的辛苦上,请友情支持一下,D瓜哥感激不尽,😜
有些打赏的朋友希望可以加个好友,欢迎关注D 瓜哥的微信公众号,这样就可以通过公众号的回复直接给我发信息。
公众号的微信号是: 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
-
target
是nums
中的一个整数 -
最多调用
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()));
}
}