【算法导论学习-012】n个数随机等概率的抽样m个
算法法导论》P129页课后题5.3-7
suppose we want to create a random sample of the set {1,2,3,…,n}, thatis, an m-element subset S, where0≤m≤n, such that each m-subset is equally likely to be created. One waywould be to set A[i]=i for i=1,2,3,…,n, call RANDOMIZE-IN-PLACE(A), and then take just the first marray elements. This method would make n calls to the RANDOM procedure. If n is much larger than m, we can create a random samplewith fewer calls to RANDOM. Show that the following recursive procedurereturns a random m-subset S of {1,2,…,n}, in which eachm-subset is equally likely, while making only m calls to RANDOM:
RANDOM-SAMPLE(m,n)
if m == 0
return ∅
else
S = RANDOM-SAMPLE(m-1, n-1)
i = RANDOM(1,n)
if i ∈ S
S = S ∪ {n}
else
S = S ∪ {i}
return S
翻译过来就是：n个数随机等概率的取样m个。
该题的证明方法1：http://clrs.skanev.com/05/03/07.html
该题的证明方法2 ：http://www.cnblogs.com/Jiajun/archive/2013/05/15/3080111.html
题目中其实给出了两种解决方案。
方案1：调用RANDOMIZE-IN-PLACE(A)
/**
* 创建时间：2014年8月13日 上午9:46:51
* 项目名称：Test
* @author Cao Yanfeng
* @since JDK 1.6.0_21
* 类说明：
*/
public class RandomSampleTest {
/**
* @param args
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] array={1,2,3,4,5,6,7,8,9,10,11,12,13,14,15};
int[] result=randomSample(array, 5);
for (int i : result) {
System.out.println(i);
}
}
public static int[] randomSample(int[] array,int m) {
randomInPlace(array);
int[] result=new int[m];
for (int i = 0; i 方案2：实现题目中的伪代码
/**
* 创建时间：2014年8月13日上午9:46:51
* 项目名称：Test
* @author Cao Yanfeng
* @since JDK 1.6.0_21
* 类说明：
*/
public classRandomSampleTest {
/**
* @param args
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] array={1,2,3,4,5,6,7,8,9,10,11,12,13,14,15};
LinkedList
for (Integer integer : result) {
System.out.println(integer);
}
}
public staticLinkedList
return sample(array, array.length, m);
}
public staticLinkedList
if (m==0) {
return newLinkedList
}else {
LinkedList
int i=array[random( 0, n-1)];
if (s.contains(i)) {
s.add(array[n-1]);
}else {
s.add(i);
}
return s;
}
}
/*返回闭区间的[a,b]随机数*/
public staticintrandom(inta,int b) {
return new Random().nextInt(b-a+1)+a;
}
}
方案3：赋予权重法
《算法导论》P1225.3节的Randomized algorithms中提供的第一种随机采样方法即赋予权重法。但是权重有可能出现相同情况，不推荐这种方法。
方案4：蓄水池抽样
见最后的扩展问题。
*******************************************************************************
正如题目中所言，如果n个数据选取m个样本，如果n远大于m，则应该使用方案2，仅调用m次random（）函数；如果n与m差距不大，则应该使用方案1,调用n次random（）函数，但是方法简单。
*******************************************************************************
扩展问题：【google面试题】给定一个数据流，其中包含无穷尽的搜索关键字（比如，人们在谷歌搜索时不断输入的关键字） 。如何才能从这个无穷尽的流中随机的选取 1000 个关键字？
参考：http://blog.csdn.net/minglingji/article/details/7984445
这也是“n个数随机等概率的取样m个”问题，但是n是未知的。采用的方式是蓄水池抽样。即：将数据流中的前1000个放入长度为1000的数组，对于1001个数，调用random(0, 1000),[0,999]闭区间内每个数被选中的概率都是1000/1001。之后对于n>1000的每个数，[0,999]闭区间内每个数被选中的概率都是1000/n。这里random被调用的次数为n-m。下面模拟一下这个过程。
/**
* 创建时间：2014年8月13日 上午9:46:51
* 项目名称：Test
* @author Cao Yanfeng
* @since JDK 1.6.0_21
* 类说明：
*/
public class RandomSampleTest {
/**
* @param args
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] array={1,2,3,4,5,6,7,8,9,10,11,12,13,14,15};
int[] result=reservoirSample(array, 5);
for (int i : result) {
System.out.println(i);
}
}
/*蓄水池抽样*/
public static int[] reservoirSample(int[] array,int m) {
int[] reservoir=new int[m];
for (int i = 0; i