蓝桥杯——印章

资源限制

内存限制：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

代码

import math

def calc_prob(n, m):
    total = pow(n, m)
    prob = 0
    for i in range(n):
        comb = math.comb(n, i)
        if i % 2 == 0:
            prob += comb * pow(n - i, m)
        else:
            prob -= comb * pow(n - i, m)
    return prob / total

n, m = map(int, input().split())
prob = calc_prob(n, m)
print("%.4f" % prob)

题解思路

首先，我们需要理解问题的背景。这是一个涉及概率和组合的问题，我们需要计算小A买了m张印章后，集齐所有n种图案印章的概率。这个问题可以借用包含-排除原理来解决。包含-排除原理是一种用于解决重叠问题的数学工具，即先包含，后排除。在这个问题中，我们先计算所有可能的情况，然后逐步排除那些不符合要求的情况。具体来说，我们首先计算所有可能的印章组合，然后减去那些缺少一种印章的组合，再加上那些缺少两种印章的组合，如此往复，直到计算完所有的情况。这就是我们在Python代码中看到的for循环部分。每次循环，我们都会计算并累加或累减特定数量的印章组合。最后，我们将计算得到的符合要求的印章组合数量除以所有可能的组合数量，就得到了所求的概率。

上面的列子可能没有完全理解，下面我举了一个列子帮大家理解

以n=2，m=3为例，我们有两种图案的印章，小A买了三张。所有可能的组合有8种（AA，AB，BA，BB，AAA，AAB，ABA，ABB）。其中，有6种组合（AB，BA，AAB，ABA，ABB，BAA）包含了所有的印章图案。所以，集齐所有图案的概率为6/8=0.75。 

大家喜欢的话就支持一下吧，大家的支持是我更新的动力~