利用蓄水池采样算法求解问题

给定一个数组，比如[‘a’,‘b’,‘a’,‘c’ ,‘b’,‘a’,‘e’,‘d’],再给定一个存在在数组中的字母比如’a’，写一个算法求一下能拿到这个字母的随机下标，比如可能是0,2,5中的随机一个，需要得出的这几个数字的下标是随机的，并且时间复杂度为O(n).

使用蓄水池的算法
int qiujie_suiji(vector<char> nums,char c)
{
	//每当a出现一次的时候就进行随机数操作过程
	int count = 0;
	int len = nums.size();
	int flag = 0;
	int xushuichi = 0;
	for (int i = 0; i < len; ++i)
	{
		if (flag == 0)//这个a没出现
		{
			if (nums[i] == c)
				xushuichi = i;
			flag = 1;
		}
		else
		{
			if (nums[i] == c)
			{
				++count;
				int rand_n = rand() % (count + 1);
				if (rand_n < 1)
					xushuichi = i;
			}
		}
	}
	return xushuichi;
}

int main()
{
	vector<char> nums = { 'a','b','a','c','a' };
	char c = 'a';
	vector<int> resulta(5,0);
	for (int i = 0; i < 100000; ++i)
	{
		auto result=qiujie_suiji(nums, c);
		//cout << result << endl;
		++resulta[result];
	}
	cout << resulta[0] << endl;
	cout << resulta[2] << endl;
	cout << resulta[4] << endl;
}