1518. 换酒问题

描述

小区便利店正在促销，用 numExchange 个空酒瓶可以兑换一瓶新酒。你购入了 numBottles 瓶酒。
如果喝掉了酒瓶中的酒，那么酒瓶就会变成空的。
请你计算 最多 能喝到多少瓶酒。
示例 1：
输入：numBottles = 9, numExchange = 3
输出：13
解释：你可以用 3 个空酒瓶兑换 1 瓶酒。
所以最多能喝到 9 + 3 + 1 = 13 瓶酒。
示例 2：
输入：numBottles = 15, numExchange = 4
输出：19
解释：你可以用 4 个空酒瓶兑换 1 瓶酒。
所以最多能喝到 15 + 3 + 1 = 19 瓶酒。
示例 3：
输入：numBottles = 5, numExchange = 5
输出：6
示例 4：
输入：numBottles = 2, numExchange = 3
输出：2
提示：
1 <= numBottles <= 100
2 <= numExchange <= 100

思路

定义几个变量：sum用来计录当前能喝到的酒的瓶数；tmp来计算喝剩下的空瓶子能换几瓶酒；remain用来计算换完酒剩下的空瓶子数目。

1.开始的时候，假设先把酒喝光，remain = numBottles即空瓶子数等于总数；
2.只要剩下的空瓶子还能兑换酒，就执行while；
3.tmp记录当前剩余的酒瓶子一共换来了几瓶酒，remain记录不足以换酒而剩下的瓶子数；
4.现在假设换来的酒又被喝完了，即执行sum += tmp，则剩余的酒瓶子数为remain += tmp。

解答

class Solution {
public:
    int numWaterBottles(int numBottles, int numExchange) {
        if (numBottles < numExchange) return numBottles;
        int sum = numBottles, tmp = 0, remain = numBottles;
        while(remain/numExchange > 0){
            tmp = remain/numExchange;//能换多少瓶酒
            remain %= numExchange;//剩下的瓶子
            sum += tmp;
            remain += tmp;
        }
        return sum;
    }
};