老鼠试药 信息熵 二进制 458. 可怜的小猪

有 1000 只水桶，其中有且只有一桶装的含有毒药，其余装的都是水。它们从外观看起来都一样。
如果小猪喝了毒药，它会在 15 分钟内死去。
问题来了，如果需要你在一小时内，弄清楚哪只水桶含有毒药，你最少需要多少只猪？
回答这个问题，并为下列的进阶问题编写一个通用算法。
进阶:
假设有 n 只水桶，猪饮水中毒后会在 m 分钟内死亡，你需要多少猪（x）就能在 p 分钟内找出 “有毒” 水桶？这 n 只水桶里有且仅有一只有毒的桶。
提示：
可以允许小猪同时饮用任意数量的桶中的水，并且该过程不需要时间。
小猪喝完水后，必须有 m 分钟的冷却时间。在这段时间里，只允许观察，而不允许继续饮水。
任何给定的桶都可以无限次采样（无限数量的猪）。

解题思路

先看简化版的问题

现有8瓶药水，其中7是完全一样的，只有一瓶里面是毒药，但是外观上分辨不出来。
毒药给小白鼠喝了后，**一小时后**这只小白鼠会突然死亡，但之前一点症状也没有。
现需要在**一小时后**找出哪瓶是毒药，问至少需要几只小白鼠？

在这一小时内，老鼠喝了药之后的状态有两种，活着或死亡。n只老鼠就可以表达 2^n 种状态。需要验证8瓶药，所以 2 ^ n >=8 ，得到n>=3；所以最少需要3只老鼠。
给药水瓶编号为000 001 010 011 100 101 110 111，
三位二进制w1、w2、w3
三只老鼠是c1、c2、c3，
老鼠c1、c2、c3与三位二进制w1、w2、w3一一对应
c3喝： 100 101 110 111(第三位是1的都喝)
c2喝： 010 011 110 111(第二位是1的都喝)
c1喝： 001 011 101 111(第一位是1的都喝)
最后假设c3死了，那么100是毒药；
假设c3c2死了，那么110是毒药；
c2c1死了，那么011是毒药，
类推，哪只老鼠死对应位上就是1，没死为0，得出的结果就是毒药瓶编号

升级版的题目

1000桶水，其中一桶有毒，猪喝毒水后会在15分钟内死去，想用一个小时找到这桶毒水，至少需要几头猪？

猪的状态改变了，在一个小时之内猪最多可以喝 60/15=4 水，对应的状态有4+1=5种。

• 在第0分钟的时候喝了一桶水以后，第15分钟死去。
• 第15分钟依然活着，喝了一桶水以后，第30分钟死去。
• 第30分钟依然活着，喝了一桶水以后，第45分钟死去。
• 第45分钟依然活着，喝了一桶水以后，第60分钟死去。
• 第45分钟依然活着，喝了一桶水以后，第60分钟依然活着。

共 5 种状态，或者可以理解为：
最多可以喝 4 次水，能够携带 base = times + 1 = 5 个的信息量，也就是（便于理解从 0 开始）：

• 喝 0 号死去，0 号桶水有毒
• 喝 1 号死去，1 号桶水有毒
• 喝 2 号死去，2 号桶水有毒
• 喝 3 号死去，3 号桶水有毒
• 喝了上述所有水依然活蹦乱跳，4 号桶水有毒

结论是 1 只小猪最多能够验证 5 桶水中哪只水桶含有毒药。而 5^n>=1000，有n>=5。
这里采用5进制对1-1000号水编号。对5只猪编号0~4。然后按照如下方法让猪喝水。

• 0号猪
  第0分钟喝位置1的数字是1的水，如图所示，也就是1、6、11、16、21…
  如果第15分钟活着，喝位置1的数字是2的水，如图所示，也就是2、7、12、17、22…
  如果第30分钟活着，喝位置1的数字是3的水，如图所示，也就是3、8、13、18、23…
  如果第45分钟活着，喝位置1的数字是4的水，如图所示，也就是4、9、14、19、24…
• 类似的，1号猪喝位置1的水…

上面，猪的编号代表5进制编码数字所在的位数，0号猪代表最末位，4号猪代表最高位。而第几分钟死代表当前位数的权重，15分钟死表示权重是1，30分钟死表示权重是2，… ，60分钟死表示权重是4，60分钟依然活着表示权重是0。
如果0号猪第30分钟死了，1号猪第15分钟死了，2号猪第45分钟死了，3，4号都活到了最后。则毒水对应的5进制编码是

原题

那么当吃药后死亡时间为minutesToDie，测试时间为minutesToTest时，一只猪在测试时间内可以喂水times = minutesToTest // minutesToDie 次，共有 base=times+1 种状态，使用base进制（在规定时间内，一只小猪最多有base种状态，就转为base进制为题），设至少需要n只猪，共有buckets瓶药，则要求base^n>=buckets ,得到n>=ceil(log(base, buckets))，向上取整。

class Solution(object):
    def poorPigs(self, buckets, minutesToDie, minutesToTest):
        """
        :type buckets: int
        :type minutesToDie: int
        :type minutesToTest: int
        :rtype: int
        """
        import math 
        times = minutesToTest // minutesToDie 
        nums = int(math.ceil(math.log(buckets, times + 1)))
        return nums