NOJ南邮上机 小明喝可乐 PROB1017 Python

PROB1017   小明喝可乐

描述：

小明是个可乐重度爱好者，平日里没事就喜欢喝可乐。今天，他最喜欢喝的崂山可乐（小明是个狠人）更新了可乐瓶换可乐活动，即每集齐 k 个可乐瓶，则赠送 1 瓶免费的可乐。
现在，小明正好手上有 n 瓶可乐，他想知道他最多能够喝多少瓶，可是他编程水平不如你，所以他想请你帮他写个程序算一下。

输入：

输入共一行，包含两个整数 n 和 k，分别表示小明手上的可乐数量和满赠活动中每多少个可乐瓶可以获得一瓶免费的可乐。

题目保证 1<n,k≤10^8 。

输出：

输出共一行一个整数，表示小明能够喝到的可乐数量。

样例输入：

10 3

样例输出： 

14

注释：

对于第一组样例，小明可以先喝 10 瓶可乐，然后拿其中 9 个可乐瓶去换 3 瓶新的可乐喝，最后再拿新的 3 瓶可乐再换一瓶新的，一共喝了 10+3+1=14瓶。

 基于python实现的代码：

def count(n, k):
    total = n
    while n >= k:
        total += n // k
        n = n // k + n % k
    return total
n,k = map(int,input().strip().split())
print(count(n, k))