蓄水池抽样算法 Reservoir Sampling

2018-03-05 14:06:40

问题描述：给出一个数据流，这个数据流的长度很大或者未知。并且对该数据流中数据只能访问一次。请写出一个随机选择算法，使得数据流中所有数据被选中的概率相等。

问题求解：如果是长度已知或者有限的问题，那么可以使用朴素的方法，先遍历一遍得到的长度。然后在得到长度后可以使用随机算法得到一个随机的index。

但是本题已经明确指出数据流长度很大或者未知，也就是说只能遍历一次，而且要保证每个数被挑选的概率相等。

标准解法是使用Reservoir Sampling算法，该算法由Knuth的学生在斯坦福读计算机博士时想出来。

算法描述：

相关问题：

382. Linked List Random Node

问题描述：

问题求解：

public class Solution {ListNode head;Random rand;/** @param head The linked list's head.Note that the head is guaranteed to be not null, so it contains at least one node. */public Solution(ListNode head) {this.head = head;this.rand = new Random();}/** Returns a random node's value. */public int getRandom() {int k = 1;ListNode cur = head;List<Integer> reservoir = new ArrayList<>();int i = 0;while (i < k && cur != null) {reservoir.add(cur.val);cur = cur.next;i++;}i++;while (cur != null) {if (rand.nextInt(i) < k) {reservoir.set(rand.nextInt(k), cur.val);}i++;cur = cur.next;}return reservoir.get(0);}
}

398. Random Pick Index

问题描述：

问题求解：

如果仅存在一个数，那么将之index返回，如果存在多个数，其index的返回值需要是等概率的，也就是说对于k个相同的数，我们需要每个index的返回概率为1/k。根据蓄水池算法，我们首先要建立一个大小为1的池子，然后对每个出现的target的index以当前出现个数的概率选择他，然后从池中随机挑选一个数来与它交换，由于池中仅有一个值，因此只需要将res的值变为挑选值即可。

public class RandomPickIndex {int[] nums;Random ran;public RandomPickIndex(int[] nums) {this.nums = nums;ran = new Random();}public int pick(int target) {int res = -1;int cnt = 0;for (int i = 0; i < nums.length; i++) {if (nums[i] != target) continue;if (ran.nextInt(++cnt) == 0) res = i;}return res;}
}

转载于:https://www.cnblogs.com/TIMHY/p/8508759.html

蓄水池抽样算法 Reservoir Sampling相关推荐

蓄水池抽样算法(reservoir sampling)
蓄水池抽样算法(reservoir sampling) 场景:在长度未知的数据流中,等概率地采样一定数量的数据.即,数据量N未知,若要求采样k个数据,采样概率保证kN\frac{k}{N}Nk. 要 ...
蓄水池抽样（Reservoir Sampling）
蓄水池抽样解决的问题是当不知道N的大小时从N个数中等概率抽取k个数. 要理解这个问题,首先从抽取一个数来看. 在知道数字总数N的大小的情况下,我们可以很容易的用C运行库的rand函数随机的获得一个数字 ...
蓄水池采样算法的python实现_蓄水池抽样算法（Reservoir Sampling）
蓄水池抽样算法(Reservoir Sampling) 许多年以后,当听说蓄水池抽样算法时,邱simple将会想起,那个小学数学老师带他做"小明对水池边加水边放水,求何时能加满水" ...
图解连续学习中的蓄水池抽样算法(The Illustrated Reservoir sampling)
图解连续学习中的蓄水池抽样算法The Illustrated Reservoir sampling 前言什么是Reservoir Sampling? 蓄水池抽样算法(Reservoir sampli ...
机器学习中的数学——蓄水池抽样算法（Reservoir Sampling Algorithm）
分类目录:<机器学习中的数学>总目录蓄水池抽样算法(Reservoir Sampling Algorithm)解决了未知长度数据的均匀抽样问题,即:给定一个数据流,数据流长度NNN很大, ...
蓄水池采样算法的python实现_常用算法-蓄水池抽样算法
Leetcode上遇到一道题,题目是这样的: 这道题的关键是链表的长度不知道,但是要使随机返回每个元素的概率相等,这一下就难倒我了,如果知道链表的长度k,从0到k中随机选择一个整数就好了呀,可现在不知 ...
水塘抽样（Reservoir sampling）
水塘抽样(Reservoir sampling) 题目:给出一个数据流,这个数据流的长度很大或者未知.并且对该数据流中的数据只能访问一次.请写出一个随机选择算法,使得数据流中所有数据被选中的概率相等. ...
LeetCode Random Pick Index(蓄水池抽样算法)
问题:给出一个数组,存在相同的数,随机输出目标数所在的下标思路:使用蓄水池抽样算法,当第一次找到目标数时,作为选取.接着如果随机数等于0,则选取.在遍历完后,直接返回选取的值具体代码参考: htt ...
【大数据算法】蓄水池抽样算法
一.题目来源: 这个题目的由来是周围有人讨论到去面试(某8)的时候遇到了这个问题.另外正好HIT有个视频也有这个内容,故记录一下: 二.题目描述: 该人面试的时候问的是: 如何从二进制文件中等 ...

蓄水池抽样算法 Reservoir Sampling

蓄水池抽样算法 Reservoir Sampling相关推荐

最新文章

热门文章