Water Jug Program In Python