OpenCV selectRandomSubset()函数源码分析

函数源码

//Returns random 'count' element subset of the {0,1,...,size-1} set
void selectRandomSubset(int count, int size, vector<int> &subset)
{
	subset.clear();
	for (int i = 0; i < size; ++i)
	{
		int x = randu<int>();
		if (x % (size - i) < count)
		{
			subset.push_back(i);
			count--;
		}
	}
}

函数作用

函数共有三个输入count，size，subset，如注释，该函数的作用就是返回{0,1,...,size-1} 数组中“count”个元素存入subset。

其中randu<int>()是以均匀分布生成int范围内的一个随机数。

//! returns the next unifomly-distributed random number of the specified type
template<typename _Tp> static inline _Tp randu() { return (_Tp)theRNG(); }

函数原理

该函数就是模拟遍历{0,1,...,size-1}集合，并对每个元素以mes_count/mes_size的概率push到subset中，因此其生成的subset是有序的。 
其中，mes_count表示还需要获取的subset元素个数，mes_size未被遍历的元素个数。

这种抽取方法可以保证每个元素都是以count/size的概率被抽取
我们知道第一个元素被抽取的概率是count/size，第二个元素被抽取的概率分两种情况：
1. 第一个元素被抽取：count-1/size-1
2. 第一个元素未被抽取：count/size-1
加上条件概率，第二个元素被抽取的概率就是(count-1/size-1)*(count/size)+(count/size-1)*(size-count/size)=count/size
后边不会推了