有两个水壶,容量分别为 jug1Capacity 和 jug2Capacity 升。水的供应是无限的。确定是否有可能使用这两个壶准确得到 targetCapacity 升。
如果可以得到 targetCapacity 升水,最后请用以上水壶中的一或两个来盛放取得的 targetCapacity 升水。
你可以:
装满任意一个水壶
清空任意一个水壶
从一个水壶向另外一个水壶倒水,直到装满或者倒空
示例 1:
输入: jug1Capacity = 3, jug2Capacity = 5, targetCapacity = 4
输出: true
解释:来自著名的 "Die Hard"
示例 2:
输入: jug1Capacity = 2, jug2Capacity = 6, targetCapacity = 5
输出: false
示例 3:
输入: jug1Capacity = 1, jug2Capacity = 2, targetCapacity = 3
输出: true
提示:
1 <= jug1Capacity, jug2Capacity, targetCapacity <= 106
思路:使用模拟法,模拟每一次操作。使用递推代替递归,因为递归栈会爆掉。
边界条件:记住每种两个杯子中的状态,使用集合进行存储。
特别提示:使用unorder_set,底层为Hash表,速度更快。但是键值对不能被Hash,需要自己编写Hash函数。
using PII = pair<int, int>;
class Solution {
public:
stack<pair<int, int>> stk;
bool canMeasureWater(int jug1Capacity, int jug2Capacity, int targetCapacity) {
stk.push(pair<int, int>(0, 0));
auto hash_function = [](const PII& o) {return hash<int>()(o.first) ^ hash<int>()(o.second); };
unordered_set<PII, decltype(hash_function)> seen(0, hash_function);
int x, y;
while (!stk.empty()) {
auto v = stk.top();
stk.pop();
x = v.first, y = v.second;
if (seen.count(PII(x, y))) {
continue;
}
if (x + y == targetCapacity || x == targetCapacity || y == targetCapacity)
return true;
seen.insert(PII(x, y));
// 把 X 壶的水灌进 Y 壶,直至Y灌满或X倒空。
stk.push(pair<int, int>(x - min(x, jug2Capacity - y), x <= jug2Capacity - y ? x + y : jug2Capacity));
// 把 Y 壶的水灌进 X 壶,直至X灌满或Y倒空。
stk.push(pair<int, int>(y <= jug1Capacity - x ? x + y : jug1Capacity, y - min(y, jug1Capacity - x)));
// 把 X 壶倒空。
stk.push(pair<int, int>(0, y));
// 把 Y 壶倒空
stk.push(pair<int, int>(x, 0));
// 把 X 壶倒满
stk.push(pair<int, int>(jug1Capacity, y));
// 把 Y 壶倒满
stk.push(pair<int, int>(x, jug2Capacity));
}
return false;
}
};