蓝桥杯 算法基础 印章

问题描述

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

输入格式

　　一行两个正整数n和m

输出格式

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

样例输入

2 3

样例输出

0.7500

数据规模和约定

　　1≤n，m≤20

题解：

这道题给出的标签是dp，说明是一道动态规划的题，那么我们就用动态规划的思维来考虑

用数组dp[i][j]来表示购买了i个印章，集齐了j个印章

接下来我们来分情况讨论

大的情况可以分为两种：

第一种：

当i<j时，即是购买印章的个数要少于集齐的印章的个数，这明显是不可能的，所以这是dp[i][j]=0;

第二种：

当i>j时，既是购买印章的个数大于集齐印章的个数，那么这时又会出现两个情况：

        当j=1时，表示的是，购买了i个印章，集齐了其中的一个印章，每个印章集齐的概率为1/n

那么购买了i次，集齐其中一个印章的概率为（1/n)^i*n, 则dp[i][j]=pow(p,i-1);

        当j！=1时，这需要来分情况讨论，第一种，购买第i个印章时，需要集齐第j个印章，说明购买的前i-1的印章已经集齐了前j-1个印章，则dp[i][j]=dp[i-1][j-1]*p*(n-j+1)

第二种，购买第i个印章时，不需要集齐第j个印章，说明购买的前i-1个印章已经集齐了j个印章，则dp[i][j]=dp[i-1][j-1]*p*j

最后将dp[m-1][n-1]输出就可以

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>

using namespace std;

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    
    double p=1.0/n;//p表示的是集齐一种印章的概率

    double dp[m+1][n+1];
    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]*p*j+dp[i-1][j-1]*p*(n-j+1);
        }

    printf("%.4lf",dp[m][n]);
    return 0;
}