蓝桥杯Python练习题18-印章

资源限制
  内存限制：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表示购买的印章的数量，则具体概率dp表示如下：（注：这里的i和j总是小于等于n的）
（1）当j<i时，即购买印章的数量＜需要集齐的印章种数时，不可能集齐印章，故符合条件的dp[i][j] = 0。

（2）当i = 1时，即需要集齐的印章种数为1时，若j = 1，则dp[i][j] = 1；若j ≠ 1，则dp[i][j] = [(1/n)]^(i-1)。这种情况下，小A购买的m张印章全部为某一种，那么第一张印章集齐那唯一一种的概率就为1，即dp[i][j] = 1；那么，当购买二张印章时，集齐i种图案的概率就变成了1/n，以此类推，dp[i][j] = [(1/n)]^(i-1)。

（3）当j > i，且i ≠ 1时，假设前几张购买的印章均不存在种类重复，而下一张印章对已经集齐的i种印章中存在重复，则dp[i][j] = P∗j/n = dp[i-1][j]∗j/n。

（4）当j > i，且i ≠ 1时，假设前几张购买的印章均不存在种类重复，而下一张印章对已经集齐的i种印章中也依然不存在重复，则dp[i][j] = P∗(n-j-1)/n = dp[i-1][j-1] ∗ (n-j-1)/n。

代码

#创建二位数组
n,m = map(int,input().split())
#将数组初始化为0
dp = [[0 for i in range(n + 1)] for j in range(m + 1)]
'''
    创建m+1行 n+1列的二维数组
    若从键盘输入4 3
    dp = [
            [0, 0, 0, 0, 0], 
            [0, 0, 0, 0, 0], 
            [0, 0, 0, 0, 0], 
            [0, 0, 0, 0, 0]
        ]
'''

for i in range(1 , m + 1):
    for j in range(1 , n + 1):
        if i < j :
            dp[i][j] = 0
        elif j == 1:
            dp[i][j] = pow(1 / n ,i - 1)
        else:
            dp[i][j] = dp[i-1][j] * (1/n) * j + dp[i-1][j-1] * (1/n)*(n-(j-1))
#打印最终结果
print("{:.4f}".format(dp[m][n]))

运行截图