蓝桥杯 饮料换购 python解法

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

a = int(input())
all = a
while(a >= 3):
    all = all + a//3  ## 这一轮喝的瓶子
    a = a//3 + a%3    ##喝完这一轮剩的盖子
print(all)