【Leetcode】1518. Water Bottles（配数学证明）

题目地址：

https://leetcode.com/problems/water-bottles/

某个商店促销，$k$个空瓶可以换一瓶水。现从商店买$n$瓶水，问最多能喝到多少瓶水。

可以这样考虑。先喝掉$n$瓶水产生$n$个空瓶。接着将这$n$个空瓶按照每组$k-1$个瓶子分组，然后从第二组借一个空瓶放到第一组，换水喝，然后再还回去，然后从第三组借一个空瓶放到第二组，换水喝，然后再还回去。进行这样的操作若干次，直到凑不出$k$个空瓶为止。我们看到，让$n$对$k-1$做带余除法，当$n=m(k-1)+r$的时候，其中$k-1>r\ge 0$。当$r=0$时，能换$m-1$瓶水，否则能换$m$瓶水，所以答案就是$$n+\lfloor \frac{n-1}{k-1}\rfloor$$其次，是否这就是最优解呢。设$x$是最优解，也就是全程喝了$x$瓶水，那么去掉一开始的$n$瓶水，剩余$x-n$个其实是用空瓶换出来的，所以被用于换水的空瓶数量就是$(x-n)k$。一共最多就$x$个空瓶，并且最后的最后还会剩余空瓶，不可能最后一瓶不剩的，所以有$$(x-n)k\le x-1$$所以$$\begin{gathered} (k-1)x\le nk-1 \\ x\le \frac{nk-n+n-1}{k-1}=n+\frac{n-1}{k-1} \end{gathered}$$$x$是整数，所以$$x\le n+\lfloor \frac{n-1}{k-1}\rfloor$$综上所述，最终答案就是$n+\lfloor \frac{n-1}{k-1}\rfloor$。

public class Solution {
    public int numWaterBottles(int numBottles, int numExchange) {
        return numBottles + (numBottles - 1) / (numExchange - 1);
    }
}

时空复杂度$O(1)$。