牛客题解 | 二项分布概率

题目

题目链接

二项分布是一种描述随机事件发生次数的概率分布，其计算公式为：

\[P(X = k) = C_n^k p^k (1-p)^{n-k} \]

其中，\(n\) 是试验次数，\(k\) 是随机事件发生的次数，\(p\) 是随机事件发生的概率。

标准代码如下

def binomial_probability(n, k, p):
    binomial_coeff = math.comb(n, k)
    probability = binomial_coeff * (p ** k) * ((1 - p) ** (n - k))
    return round(probability, 5)