Craking the Coding Interview 面试题：随机抽出子数列

于 2014-01-31 09:48:00 发布 · 158 阅读

本文介绍了一种从大数列中随机抽取子数列的算法，确保每个元素被选中的概率相同。通过使用C++实现，演示了如何从一个整数数组中随机选择固定数量的元素。

在给定的一个大数列中，随机抽出一个子数列，要求所有数被抽出的概率都是一样的，如下面的运行结果：

int rangeRandNum(int a, int b)
{
	return rand()%(b-a+1) + a;
}


vector<int> randPickArray(vector<int> &arr, int m)
{
	vector<int> picks(arr.begin(), arr.begin()+m);
	for (int i = m; i < arr.size(); i++)
	{
		int t = rangeRandNum(0, i);
		if (t < m)
		{
			picks[t] = arr[i];
		}
	}
	return picks;
}


int main() 
{
	int tar = 7;
	int cand[] = {1,2,3,0,3,2,0,3,1,4,5,3,2,7,5,3,0,1,2,1,3,4,6,8,1,8};

	srand(time(time_t(NULL)));

	vector<int> v(cand, cand+sizeof(cand)/sizeof(int));
	for (auto x:v)
		cout<<x<<" ";
	cout<<endl;

	vector<int> r = randPickArray(v, 6);

	for (auto x:r)
		cout<<x<<" ";
	cout<<endl;

	system("pause"); 
	return 0;
}