Post

1642 Water Bottles

1642 Water Bottles

Water Bottles image

There are numBottles water bottles that are initially full of water. You can exchange numExchange empty water bottles from the market with one full water bottle.

The operation of drinking a full water bottle turns it into an empty bottle.

Given the two integers numBottles and numExchange, return the maximum number of water bottles you can drink.

 

Example 1:

image

1
2
3
4
5
6
**Input:** numBottles = 9, numExchange = 3
**Output:** 13
**Explanation:** You can exchange 3 empty bottles to get 1 full water bottle.
Number of water bottles you can drink: 9 + 3 + 1 = 13.

Example 2:

image

1
2
3
4
5
6
**Input:** numBottles = 15, numExchange = 4
**Output:** 19
**Explanation:** You can exchange 4 empty bottles to get 1 full water bottle. 
Number of water bottles you can drink: 15 + 3 + 1 = 19.

 

Constraints:

1
2
1 <= numBottles <= 100
2 <= numExchange <= 100
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

class Solution:
    def numWaterBottles(self, numBottles: int, numExchange: int) -> int:
        res = numBottles
        leftover = 0
        while(numBottles >= numExchange):
            print(numBottles)
            leftover = numBottles % numExchange
            numBottles = numBottles // numExchange
            res += numBottles
            numBottles += leftover
        return res



This post is licensed under CC BY 4.0 by the author.