518. Water Bottles

class Solution:
    def numWaterBottles(self, numBottles: int, numExchange: int) -> int:
        rem = numBottles      
        drunk = numBottles
        emptyBottles = numBottles

        while emptyBottles >= numExchange:
            exchangedBottles = emptyBottles // numExchange
            emptyBottles = emptyBottles % numExchange
            drunk += exchangedBottles
            emptyBottles += exchangedBottles

        return drunk