Water Jug Problem.py

# Implementation of Water Jug Problem in Python.
# By: Darpit Makwana

# Import defaultdict from collections library
from collections import defaultdict

# Get Values From User
jug1 = int(input("Enter Maximum capacity of Jug 1 : "))
jug2 = int(input("Enter Maximum capacity of Jug 2 : "))
aim = int(input("Enter Goal value of Jug : "))

visited = defaultdict(lambda: False)

def waterJugSolver(amt1, amt2):
    if (amt1 == aim and amt2 == 0) or (amt2 == aim and amt1 == 0):
        print(amt1, amt2)
        return True

    if visited[(amt1, amt2)] == False:
        print(amt1, amt2)

        visited[(amt1, amt2)] = True

        return (waterJugSolver(0, amt2) or
                waterJugSolver(amt1, 0) or
                waterJugSolver(jug1, amt2) or
                waterJugSolver(amt1, jug2) or
                waterJugSolver(amt1 + min(amt2, (jug1 - amt1)),
                               amt2 - min(amt2, (jug1 - amt1))) or
                waterJugSolver(amt1 - min(amt1, (jug2 - amt2)),
                               amt2 + min(amt1, (jug2 - amt2))))

    else:
        return False

print("\nS T E P S : \n-----------")

# Run Function with initial values (0,0) bacause initially both jugs are empty.
waterJugSolver(0, 0)