蓝桥杯——ALGO-1007——印章

题目描述如下：

问题描述

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

输入格式

　　一行两个正整数n和m

输出格式

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

样例输入

2 3

样例输出

0.7500

解法：

看着dp就感觉很难，参考完解析发现是可以想通的。自己做这样的题目，慢慢摸索出了技巧。

技巧就是：一支笔和一张草稿纸

也看到递归的解法，不够这里讲的是循环。

dp的状态转换可以看成是当印章数量i和图案种类j变化时，对应当前状态下集齐所以图案的概率dp。

• 首先，我们要明确符号代表的含义：
• m，n表示输入的m个印章和需要收集的n种图案
• dp[i][j] 表示第i个印章和已经收集j种图案时，完成收集全部n种图案的概率
• n个印章类别，期望p=1/n

好的，可以对问题进行分情况讨论了

1. i<j:    当印章数i比图案种类j小的时候，说明。。。说明不可能收集n种图案，即dp=0（很容易想通）
2. j==1 当图案种类就1种时，有两种情况：
   1. i==1 当印章数也等于1时，一个印章对应一个图案。此刻的dp肯定是1啊，即dp=1
   2. i>1 当印章数大于1时，这个就有点点绕人了，先上结论，dp= = .因为你要保证印章数大于1且图案种数也为一，说明全部都是一种图案，比如都是A，所以概率都是p。 i个数概率自然是。都是B呢？，自然还是。都是C呢，D呢？都是。总共几种可能？ 答案是n种。所以dp= =
3. 其他情况，  就可以分为两大类了
   1. dp[i][j]中 第j个印章代表的图案已经出现过，比如抽奖抽到之前抽中的商品。此时.    dp[i][j]=dp[i-1][j](没抽奖之前的概率)*(p*j)(抽奖抽中之前抽过的商品的概率)
   2. dp[i][j]中 第j个印章代表的图案没有出现过，比如抽奖抽到新的商品。此时                       dp[i][j]=dp[i-1][j-1](没抽奖之前的概率)*(p*n-(j-1))(抽奖抽中新商品的概率)

代码中可以发现 2.1和2.2的情况被一个公式合并了

dp[i][j]=p**(i-1)

是因为。一个数的0次方========1

直接上代码吧！

代码：

n,m=map(int,input().split())
p=1/n
# i个印章,j种图案，  概率dp
dp=[[0 for j in range(n+1)] for i in range(m+1)]
for i in range(1,m+1):
    for j in range(1,n+1):
        if i<j:
            dp[i][j]=0
        if j==1:
            dp[i][j]=p**(i-1)
        else:
            dp[i][j]=dp[i-1][j]*(j*p)+dp[i-1][j-1]*p*(n-(j-1))
print("{:.4f}".format(dp[m][n]))