题意

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

注意：

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

示例:

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

题解

蓄水池抽样问题

首先从最简单的例子出发：数据流只有一个数据。我们接收数据，发现数据流结束了，直接返回该数据，该数据返回的概率为1。

看来很简单，那么我们试试难一点的情况：假设数据流里有两个数据。我们读到了第一个数据，这次我们不能直接返回该数据，因为数据流没有结束。我们继续读取第二个数据，发现数据流结束了。因此我们只要保证以相同的概率返回第一个或者第二个数据就可以满足题目要求。因此我们生成一个0到1的随机数R，如果R小于0.5我们就返回第一个数据，如果R大于0.5，返回第二个数据。

接着我们继续分析有三个数据的数据流的情况。为了方便，我们按顺序给流中的数据命名为1、2、3。我们陆续收到了数据1、2和前面的例子一样，我们只能保存一个数据，所以必须淘汰1和2中的一个。应该如何淘汰呢？不妨和上面例子一样，我们按照二分之一的概率淘汰一个，例如我们淘汰了2。继续读取流中的数据3，发现数据流结束了，我们知道在长度为3的数据流中，如果返回数据3的概率为1/3,那么才有可能保证选择的正确性。也就是说，目前我们手里有1,3两个数据，我们通过一次随机选择，以1/3的概率留下数据3，以2/3的概率留下数据1。那么数据1被最终留下的概率是多少呢？

数据1被留下：(1/2) * (2/3) = 1/3
数据2被留下概率：(1/2) * (2/3) = 1/3
数据3被留下概率：1/3

这个方法可以满足题目要求，所有数据被留下返回的概率一样！

因此，我们做一下推论：

假设当前正要读取第n个数据，则我们以 1/n 的概率留下该数据，否则留下前 n-1 个数据中的一个。以这种方法选择，所有数据流中数据被选择的概率一样。

//C++
class Solution {private:vector<int> nums;
public:Solution(vector<int>& nums) {for(int num : nums){this->nums.push_back(num);}}int pick(int target) {int index = 0;int n = 0;for(int i = 0; i < nums.size(); i++){if(nums[i] == target){n++;if(rand() % 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);*/

LeetCode系列398—随机数索引（蓄水池抽样）相关推荐

398. 随机数索引（哈希表预处理 Or 蓄水池抽样）
文章目录 Question Ideas 1.Answer( Java ) `⚡️ getOrDefault(Object key, V defaultValue)` Code①( HashMap 实现 ...
leetcode刷题记录-398. 随机数索引
前言今天的题目为中等,跟之前碰到过的一道题思路很相似,利用map表来空间换时间,以此来做到节省时间复杂度. 每日一题今天的题目是 398. 随机数索引,难度为中等给你一个可能含有重复元素的整 ...
Java实现 LeetCode 398 随机数索引
398. 随机数索引给定一个可能含有重复元素的整数数组,要求随机输出给定的数字的索引. 您可以假设给定的数字一定存在于数组中. 注意: 数组大小可能非常大. 使用太多额外空间的解决方案将不会通过测试 ...
398. 随机数索引（设计）
LeetCode:398. 随机数索引要求返回 target 在数组中随机的一个下标留意,这个数组不是有序的,有序的话使用二分查找. 这里无序的, 直接保存相同元素的下标,然后随机返回 AC C ...
LeetCode 398. 随机数索引（概率）
1. 题目给定一个可能含有重复元素的整数数组,要求随机输出给定的数字的索引. 您可以假设给定的数字一定存在于数组中. 注意: 数组大小可能非常大. 使用太多额外空间的解决方案将不会通过测试. 示例: ...
LeetCode 398. 随机数索引
1.题目 https://leetcode-cn.com/problems/random-pick-index/ 2.题解蓄水池抽样其思路就是循环之后先判断一下这个数在数组中的索引, 然后随机选出 ...
Leetcode 398. 随机数索引解题思路及C++实现
解题思路: 题目对空间复杂度有要求.在构造函数中,应该要先把nums数组拷贝一下. 在pick函数中,必然要遍历数组nums,这样才能得到 target 值对应的所有索引,之后再生成一个随机数,返回任 ...
Leetcode 398.随机数索引
原题链接:Leetcode 398. Random Pick Index Given an integer array nums with possible duplicates, randomly ...
leetcode 398. Random Pick Index | 398. 随机数索引（Java）
题目 https://leetcode.com/problems/random-pick-index/ 题解常规思路,先用 map 存一串,取的时候从 map 对应的串中随机拿一个就 ok. cla ...

LeetCode系列398—随机数索引（蓄水池抽样）

题意

题解

LeetCode系列398—随机数索引（蓄水池抽样）相关推荐

最新文章

热门文章