这道题我一开始想用概率来做,但怎么都不ac,最后参考了一下别人的思路。(动态规划)
话不多说,上代码:
//共有n种图案的印章,每种图案的出现概率相同。小A买了m张印章,求小A集齐n种印章的概率。
//输入两个整数n和m
//输出实数P保留4位小数
//动态规划
// 两个变量二维数组dp[i][j](买i种凑j种)
//初始状态:
// i<j,dp[i][j]=0
//j==1,dp[i][1]=pow(1/n,i)*n(n种可能)即:dp[i][1]=pow(1/n,i-1)
//中间状态:dp[i][j]
// 两种可能:第i张对比第i-1有重复和没重复
//有重复p=dp[i-1][j]*(j/n),第i次抽到j种里概率为(j/n)
//无重复p=dp[i-1][j-1]*((n-(j-1))/n),第i次抽到除j-1种里概率为((n-(j-1))/n)
//两种可能相加dp[i][j]=p=dp[i-1][j]*(j/n)+p=dp[i-1][j-1]*((n-(j-1))/n)
#include<iostream>
#include<iomanip>
#include<math.h>
using namespace std;
double dp[22][22];
int main() {
int n, m;
cin >> n >> m;
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][1] = pow(p, i - 1);
else dp[i][j] = 1.0 * j / n * dp[i - 1][j] + 1.0 * (n - (j - 1)) / n * dp[i - 1][j - 1];
cout << setprecision(4) << fixed << dp[m][n] << endl;
return 0;
}