蓝桥杯python：饮料换购

题目：
程序说明：
将购买的数量不断整除3，将除得的数加上余数之后再整除3，直到所剩的数量不足3瓶。
全部代码：

n=int(input())
sum=n
while n>=3:
    a=n//3
    sum+=a
    n=a+n%3
print(sum)