有 buckets 桶液体,其中 正好 有一桶含有毒药,其余装的都是水。它们从外观看起来都一样。为了弄清楚哪只水桶含有毒药,你可以喂一些猪喝,通过观察猪是否会死进行判断。不幸的是,你只有 minutesToTest 分钟时间来确定哪桶液体是有毒的。
喂猪的规则如下:
选择若干活猪进行喂养
可以允许小猪同时饮用任意数量的桶中的水,并且该过程不需要时间。
小猪喝完水后,必须有 minutesToDie 分钟的冷却时间。在这段时间里,你只能观察,而不允许继续喂猪。
过了 minutesToDie 分钟后,所有喝到毒药的猪都会死去,其他所有猪都会活下来。
重复这一过程,直到时间用完。
给你桶的数目 buckets ,minutesToDie 和 minutesToTest ,返回在规定时间内判断哪个桶有毒所需的 最小 猪数。
示例 1:
输入:buckets = 1000, minutesToDie = 15, minutesToTest = 60
输出:5
示例 2:
输入:buckets = 4, minutesToDie = 15, minutesToTest = 15
输出:2
示例 3:
输入:buckets = 4, minutesToDie = 15, minutesToTest = 30
输出:2
提示:
1 <= buckets <= 1000
1 <= minutesToDie <= minutesToTest <= 100
题目来源:LeetCode 458. 可怜的小猪
思路
最多进行k
轮试验(总时间/每次试验的时间)
设水桶数量n
猪数量x
每个猪可能不死、1
轮后死、2
轮后死…k
轮后死 一共k+1
种状态
x
只猪的总状态数量为
(
k
+
1
)
x
(k+1)^x
(k+1)x
总状态数量需大于等于n
即
(
k
+
1
)
x
≥
n
(k+1)^x≥n
(k+1)x≥n
由此得出x
如何构造方案?
将每桶水编号0 ~ n-1
转换为k+1
进制数 猪的编号为0 ~ x-1
若第i
位上的数为a[i]
表示第a[i]
轮喂第i
头猪 a[i]
为0
则表示不喂
若第y
桶水有毒 其k+1
进制表示为
y
=
(
a
0
a
1
.
.
.
a
x
−
1
)
k
+
1
y=(a_0a_1...a_{x-1})_{k+1}
y=(a0a1...ax−1)k+1
若
a
i
≠
0
a_i≠0
ai=0 则第
i
i
i头猪会在
a
i
a_i
ai轮死去
观察猪的状态可以反推出y
故存在可行的方案
即
x
=
log
k
+
1
n
x=\log_{k+1}n
x=logk+1n
思路来源:acwing
代码
class Solution {
public:
int poorPigs(int buckets, int minutesToDie, int minutesToTest) {
int k = minutesToTest / minutesToDie;
return ceil(log(buckets) / log(k + 1));
}
};