LeetCode 458. 可怜的小猪（C++）

有 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$
总状态数量需大于等于n
即$(k+1{)}^x\ge 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}$
若$a_i\ne 0$ 则第$i$头猪会在$a_i$轮死去
观察猪的状态可以反推出y
故存在可行的方案
即$x={\log }_{k+1}n$
思路来源：acwing

代码

class Solution {
public:
    int poorPigs(int buckets, int minutesToDie, int minutesToTest) {
        int k = minutesToTest / minutesToDie;
        return ceil(log(buckets) / log(k + 1));
    }
};