最简洁的解

参考代码:

n = int(input())

# 初始饮料数和瓶盖数
total_drinks = n
caps = n

# 不断循环，直到无法再换取新的饮料
while caps >= 3:
    # 可以换取的饮料数
    drinks_from_caps = caps // 3
    # 更新剩余瓶盖数
    caps = caps % 3 + drinks_from_caps
    # 更新总饮料数
    total_drinks += drinks_from_caps

# 输出总饮料数
print(total_drinks)