题目链接
458. 可怜的小猪 - 力扣(LeetCode) (leetcode-cn.com)
题目概述
有 buckets 桶液体,其中 正好 有一桶含有毒药,其余装的都是水。它们从外观看起来都一样。为了弄清楚哪只水桶含有毒药,你可以喂一些猪喝,通过观察猪是否会死进行判断。不幸的是,你只有 minutesToTest 分钟时间来确定哪桶液体是有毒的。
喂猪的规则如下:
选择若干活猪进行喂养
可以允许小猪同时饮用任意数量的桶中的水,并且该过程不需要时间。
小猪喝完水后,必须有 minutesToDie 分钟的冷却时间。在这段时间里,你只能观察,而不允许继续喂猪。
过了 minutesToDie 分钟后,所有喝到毒药的猪都会死去,其他所有猪都会活下来。
重复这一过程,直到时间用完。
给你桶的数目 buckets ,minutesToDie 和 minutesToTest ,返回在规定时间内判断哪个桶有毒所需的最小猪数。
解题思路
第一反应这题的猪和桶有数学关系,但实在想不出来(主要是觉得困难题不能这么简单)。
动态规划:反向思考,题目求固定buckets下猪的最小值,可以求出不同数量的猪对应buckets的最大值。
边界条件:当buckets=1时,不需要猪;当猪或者测试次数=0时,只能有唯一的毒牛奶,即buckets=0;
转移公式:;其中i表示猪的数目,j表示测试次数,f表示i头猪在j轮测试下可以测出毒牛奶的最大buckets。
class Solution {
public:
int poorPigs(int buckets, int minutesToDie, int minutesToTest) {
if(buckets == 1){
return 0;
}
int n = minutesToTest/minutesToDie;
vector<vector<int>> dp(buckets+1,vector<int>(buckets+1));
vector<vector<int>> Com(buckets+1,vector<int>(buckets+1));
for(int i=0;i<buckets+1;i++){
dp[i][0] = 1;
dp[0][i] = 1;
Com[i][i] = 1;
Com[0][i] = 1;
}
for(int i=1;i<buckets+1;i++){
for(int j=1;j<i;j++){
Com[j][i] = Com[j-1][i-1] + Com[j][i-1];
}
for(int j=1;j<n+1;j++){
for(int k=0;k<=i;k++){
dp[i][j] += dp[k][j-1]*Com[k][i];
}
if(dp[i][j] >= buckets){
return i;
}
}
}
return 0;
}
};
官方题解
可怜的小猪 - 可怜的小猪 - 力扣(LeetCode) (leetcode-cn.com)
思路1为动态规划,同思路;
思路2为数学方法,评论有补充。