蓝桥杯 算法训练 印章

问题描述

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

输入格式

一行两个正整数n和m。

数据规模和约定：1≤n，m≤20。

输出格式

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

示例

样例输入

2 3

样例输出

0.7500

AC code

/// 试题 算法训练 印章
#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n, m;
    cin>>n>>m;

    double dp[m+1][n+1] = {0.000000};

    double p = 1.0/n;

    for(int i=1; i<=m; i++)
        dp[i][1] = pow(p, i-1);

    for(int j=2; j<=n; j++)
        for(int i=2; i<=m; i++)
            dp[i][j] = dp[i-1][j-1]*(n-j+1)*1.0/n + dp[i-1][j]*j*1.0/n;

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

    return 0;
}