PYTHON答题洛谷P1150 PETER的烟

题目描述

Peter 有 𝑛n 根烟，他每吸完一根烟就把烟蒂保存起来，𝑘k（𝑘>1k>1）个烟蒂可以换一个新的烟，那么 Peter 最终能吸到多少根烟呢？

与某些脑筋急转弯不同的是，Peter 并不能从异次元借到烟蒂，抽完后再还回去。

输入格式

每组测试数据一行包括两个整数 𝑛,𝑘n,k（1<𝑛,𝑘≤1081<n,k≤108）。

输出格式

对于组测试数据，输出一行包括一个整数表示最终烟的根数。

参考代码：

n,k=map(int,input().split())
sum=n
if n<k:
    print(n)
while n>=k:
    sum+=n//k
    n=n//k+n%k   
print(sum)