蓝桥杯 算法训练 印章

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

---

问题描述

　　共有n种图案的印章，每种图案的出现概率相同。小A买了m张印章，求小A集齐n种印章的概率。

输入格式

　　一行两个正整数n和m

输出格式

　　一个实数P表示答案，保留4位小数。

样例输入

2 3

样例输出

0.7500

数据规模和约定

　　1≤n，m≤20

思路：

        这个题看着很简单，但是其实还是要搞懂一些数学知识的，后面再发现是个动态规划题。我们首先定义dp[i][j]表示第i次购买到j种印章的概率，那么求到最后dp[m][n]也就是第m次购买到n种印章的概率；注意：我们这里的第i次购买到j种，并不是正好购买到j种，包含了前面已经购买了j种的情况，下面给出三种情况并解释：

if(i<j)dp[i][j]=0;
else if(j==1)dp[i][j]=pow(p,i-1);

else{
        dp[i][j] = dp[i-1][j]*j*p+dp[i-1][j-1]*p*(n-(j-1));
 }

第一种情况：

        i<j;也就是买的还没种类多，那么肯定凑不齐概率为零；

第二种情况：

        j = 1；也就是你购买到第i次了还只是购买到一种印章，说明前面一直重复购买一种印章，由于购买到每种印章的概率都相等，都是1/n = p，所以前面i-1次都是购买同一种印章的概率就是(i-1)^p;

第三种情况：

首先我们应该知道     本轮状态概率 = 上一个状态的概率 * 此次购买种类概率        

        有两种选择，要么此次购买的种类已经在前面购买过了，那么上一轮的概率就是dp[i-1][j]，此次购买种类是j种，j*p就是再次买到购买过的j种印章概率，合在一起就是 dp[i-1][j]*j*p；

        如果此次购买的种类之前并没有买过，那么上一次只能能达到j-1种印章数（因为本次循环是dp[i][j],上次没达到j，只能是j-1咯）那么上一轮的概率就是dp[i-1][j-1]，此次购买种类是n-(j-1)种，n-(j-1)*p就是买到未购买过的第j种印章概率，合在一起就是 dp[i-1][j-1]*n-(j-1)*p；

第三种情况就是以上两种选择合在一起，所以是dp[i][j] = dp[i-1][j]*j*p+dp[i-1][j-1]*p*(n-(j-1));

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 20+10;
int n,m,t,d;
int a[N],b[N];
string s; 
signed main(){
	cin>>n>>m;
	double dp[N][N];
	double p = 1.0/n;
	for(int i=1;i<=m;i++){
		for(int j=1;j<=n;j++){
			if(i<j)dp[i][j]=0;
			else if(j==1)dp[i][j]=pow(p,i-1);
			else{
				dp[i][j] = dp[i-1][j]*j*p+dp[i-1][j-1]*p*(n-(j-1));
			}
		}
	}
	printf("%.4f",dp[m][n]); 
	
	return 0;
}