k = int(input())
p = k
i = 1
while p >= i:
    p -= i
    i += 1
i -= 1
q = int()
print(int(i * (i + 1) * (2 * i + 1) / 6 + (i + 1) * (k - i * (i + 1) / 2)))

自然数平方和n(n+1)(2n+1)/6了解一下。

我的思路是把金币分成两部分，第一部分全部都是连续n天每天获得n个金币，后一部分是剩余不足n天每天获得n个金币，分别计算。