给一个无限长链表，怎么随机抽K个数，保证每个数抽到概率一致——蓄水池抽样

假老练啊哦

于 2019-09-05 14:56:53 发布

阅读量694

点赞数

分类专栏：面经中遇到问题

原文链接：https://blog.youkuaiyun.com/huagong_adu/article/details/7619665#commentBox

版权

面经中遇到问题专栏收录该内容

40 篇文章

订阅专栏

本文介绍了一种从大量数据中随机抽取固定数量样本的算法——蓄水池抽样，确保每个元素被选中的概率相等。通过《编程珠玑》中的案例，详细解析了算法实现原理及代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

类似：数据流，选K个数，怎么保证每个数抽取概率一样？

转载：蓄水池抽样——《编程珠玑》读书笔记

int random_pick(int m) {
	int ret = rand() % (m + 1);
	return ret;
}
void reservoir_sample(int* input, int * result, int k,int n ) {
	srand(time(NULL));
	for (int i = 0; i < k; i++) {
		result[i] = input[i];
	}
	for (int j = k; j < n; j++) {
		int pos = random_pick(j);
		if (pos < k)
			result[pos] = input[j];
	} 
}
int main(){
	int *input = new int[100 * sizeof(int)];
	int *result = new int[10 * sizeof(int)];
	reservoir_sample(input, result, 10,100);
	for (int i = 0; i < 10; i++)	cout << input[i] << " ";
}