365. 水壶问题

有两个水壶,容量分别为 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;
    }
};

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值