题目描述
假设有k只小鸟和n个树洞,每只小鸟均匀随机选择一个树洞,给定n,计算当k不小于多少时,至少存在一对小鸟选择了同一个树洞这件事的概率不小于p(0<p<=1)。
输入输出描述
输入n(1<=n<=10000),p(0<p<=1),分别为树洞的数量和概率值。
365 0.5
输出整数m,当k>=m时,至少存在一对小鸟选择了同一个树洞的概率>=p。
23
说明:
假设有365个树洞,当小鸟的数量不小于23时,至少存在一对小鸟选择同一个树洞的概率不小于0.5
思路及代码
题目所求事件的对立事件为所有小鸟都进不同洞,其概率target=1-p,当前小鸟都进不同洞的概率大于target时,至少有两只鸟进了同一个树洞。
n, p = list(input().strip().split())
n = int(n)
p = float(p)
product = 1
target = 1 - p
res = 0
while product > target:
product *= (n - res) / n
res += 1
print(res)