好简单

最终代码：

# 读取输入的天数k
k = int(input())

total = 0  # 总金币数
current_day = 0  # 已经计算的天数
n = 1  # 当前阶段每天的金币数

# 循环计算每个阶段
while current_day < k:
    # 如果剩下的天数 >= 当前阶段的天数
    if current_day + n <= k:
        total += n * n
        current_day += n
    else:
        # 剩下的天数不足一个阶段
        remain = k - current_day
        total += remain * n
        current_day = k  # 结束循环
    n += 1  # 进入下一个阶段

print(total)