从1-N的数字里选出M个数字组成一个数组，要求每个数字在每个位置的期望相同

最新推荐文章于 2021-01-21 15:02:07 发布

原创最新推荐文章于 2021-01-21 15:02:07 发布 · 914 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#洗牌算法 #从1-N中选M个数字

收录于

秋招专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一种使用洗牌算法从1到N中随机选取M个数字的方法，确保了每个数字出现在数组中每个位置的期望相同。通过Knuth-Durstenfeld Shuffle算法实现了数组的完全随机排列。

题目：从1-N的数字里选出M个数字组成一个数组，要求每个数字在每个位置的期望相同

1.洗牌算法思想

//1.思路为借助洗牌算法的完全随机性，选其中的M形成数组
vector<int> NewArray(int n, int m)
{
	if (n < m)
		return vector<int>();

	vector<int> base(n);
	for (int i = 1; i <= n; i++)
		base[i - 1] = i;
	//以下为著名的Knuth-Durstenfeld Shuffle 洗牌算法，注意取余为%(i+1)
	for (int i = n - 1; i >= n - m; i--)
	{
		srand((unsigned)time(NULL));
		swap(base[i], base[rand() % (i + 1)]);
	}
	vector<int> save(m);
	for (int i = n - m, j = 0; i <= n - 1; i++, j++)
		save[j] = base[i];
	return save;
}