蓝桥杯：饮料换购--Python解法

问题描述：

乐羊羊饮料厂正在举办一次促销优惠活动。乐羊羊C型饮料，凭3个瓶盖可以再换一瓶C型饮料，并且可以一直循环下去(但不允许暂借或赊账)。
请你计算一下，如果小明不浪费瓶盖，尽量地参加活动，那么，对于他初始买入的n瓶饮料，最后他一共能喝到多少瓶饮料。

输入：

一个整数n，表示开始购买的饮料数量（0<n<10000）

输出：

一个整数，表示实际得到的饮料数

例如：

用户输入：

100

程序应该输出：

149

用户输入：

101

程序应该输出：

151

资源约定：

峰值内存消耗 < 256M
CPU消耗 < 1000ms

代码：

def run():
    count = int(input())
    bottle_cap = count

    while bottle_cap >= 3:
        count += bottle_cap // 3
        bottle_cap = bottle_cap // 3 + bottle_cap % 3
    print(count)
run()