365. Water and Jug Problem

Medium (Trung bình) Python 🔗 Xem trên LeetCode

📋 Đề Bài

You are given two jugs with capacities jug1Capacity and jug2Capacity liters. There is an infinite amount of water supply available. Determine whether it is possible to measure exactly targetCapacity liters using these two jugs.

If targetCapacity liters of water are measurable, you must have targetCapacity liters of water contained within one or both buckets by the end.

Operations allowed:

  • Fill any of the jugs with water.
  • Empty any of the jugs.
  • Pour water from one jug into another till the other jug is completely full, or the first jug itself is empty.

 

Example 1:

Input: jug1Capacity = 3, jug2Capacity = 5, targetCapacity = 4
Output: true
Explanation: The famous Die Hard example

Example 2:

Input: jug1Capacity = 2, jug2Capacity = 6, targetCapacity = 5
Output: false

Example 3:

Input: jug1Capacity = 1, jug2Capacity = 2, targetCapacity = 3
Output: true

 

Constraints:

  • 1 <= jug1Capacity, jug2Capacity, targetCapacity <= 106

💻 Lời Giải

Python 0365-water-and-jug-problem.py 
class Solution:
    def canMeasureWater(self, jug1Capacity: int, jug2Capacity: int, targetCapacity: int) -> bool:
        waters = [jug1Capacity, -jug1Capacity, jug2Capacity, -jug2Capacity]
        limit = jug1Capacity + jug2Capacity
        dq = deque([0])
        visited = set([0])
        
        while dq:
            capacity = dq.popleft()
            
            if capacity == targetCapacity:
                return True
            
            for water in waters:
                newCapacity = capacity + water
                if 0 <= newCapacity <= limit and newCapacity not in visited:
                    dq.append(newCapacity)
                    visited.add(newCapacity)
                    
        return False