2015年蓝桥杯A组C/C++ 第八题饮料换购python实现

问题描述
乐羊羊饮料厂正在举办一次促销优惠活动。乐羊羊C型饮料，凭3个瓶盖可以再换一瓶C型饮料，并且可以一直循环下去(但不允许暂借或赊账)。

请你计算一下，如果小明不浪费瓶盖，尽量地参加活动，那么，对于他初始买入的n瓶饮料，最后他一共能喝到多少瓶饮料。

输入：一个整数n，表示开始购买的饮料数量（0<n<10000）
输出：一个整数，表示实际得到的饮料数

例如：
用户输入：
100
程序应该输出：
149

用户输入：
101
程序应该输出：
151

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 1000ms

Code：

n = int(input())
res = n
m = n
while m >= 3:
    i = m // 3
    res += i
    m = m % 3+i
print(res)