蓝桥杯练习题ALGO-1007 印章

问题描述

共有n种图案的印章，每种图案的出现概率相同。小A买了m张印章，求小A集齐n种印章的概率。
输入格式
　　一行两个正整数n和m
输出格式
　　一个实数P表示答案，保留4位小数。
样例输入
2 3
样例输出
0.7500
数据规模和约定
　　1≤n，m≤20

算法分析

1 构建数组存储元素，这里的元素为概率，与题目相关的有抽的次数和抽中的印章数目两种情况，因此构建一个（m+1）*（n+1）的数组，数组是从0开始的，为了方便，不使用下标m=0的情况，因此开一个（m+1）(n+1)的数组。m表示抽的次数，n是集齐的印章数目。

2 确定转移方程
先从一个简单的实例来分析

图中我用蓝色标记为1号的格子的概率，抽3中2，有两种情况，一种是前两次抽了同一种徽章(概率存储在1号格子中)，第三次抽了其他的任意一个徽章（概率为未抽到的徽章n-(j-1)除于总徽章），对应的方程为arr[2][1]*（2-1）\2。第二种情况为前两次抽到两个不同的徽章，第三次抽中已经抽中的徽章中的任意一个徽章（概率为1/），对应方程arr[2][2]*1/2。更为一般的我们得出方程
arr[i][j] = arr[i - 1][j - 1] * (n - j+1) / n + arr[i - 1][j] * j / n;

3 确定特殊情况。当抽取次数小于集齐的徽章个数，显然概率为0。当集齐的种类为1时，情况为抽中任意一个，剩下的每次都抽中同一个类型的徽章。概率为1/n的i-1次方。

4 得出解，要求n个勋章，抽m次，即为输出结果arr[m][n]。

#include<iostream>
#include<cmath>
#include<iomanip>
using namespace std;

void probability(int m, int n);
int main()
{
	int n, m;
	cin >> n;
	cin >> m;
	probability(n, m);
}
void probability(int n, int m)
{
	//动态开辟二维数组存储抽到的概率
	//m为抽的次数，n为集齐的硬币数量
	double** arr = new double* [m+1];
	for (int i = 0; i <= m; i++)
	{
		arr[i] = new double[n+1];
	}

	for (int i = 1; i < m+1; i++)
	{
		for (int j = 1; j < n+1; j++)
		{
			//当抽的次数小于抽中的次数，概率为0
			if (i<j)
			{
				arr[i][j] = 0;
			}
			//只抽中一种勋章的概率
			else if (j==1)
			{
				arr[i][j] = pow(1.0 / n, i-1);
			}
			else
			{
				//其他概率
				//当前的格子的概率应该是它上一个格子抽中的概率*再重复抽中已经抽中的格子中的印章的概率+它左上方的格子再抽中一个未抽中的印章的概率
				arr[i][j] = arr[i - 1][j - 1] * (n - j+1) * 1.0 / n + arr[i - 1][j] * (j * 1.0) / n;
			}
		}
	}
	
	cout <<fixed<<setprecision(4)<< arr[m][n];
}

提交结果