天数 1 2 3 4 5 6 7 ...

金额 1 2 2 3 3 3 4 ...

找规律: 1天1 2天11+2 3天11+22 6天11+22+33 7天11+22+33+41 可以看到3天,6天刚好为天数相加,同时为价格平方相加 7天时候是6天的价格+ (7-6)*4 11天的时候是10天的价格 + (11-(1+2+3+4))*5

k = int(input())
su = 0 
count = 0 # 总金额
if k == 1:
    print(1)
else:
    for i in range(1,k+1):#对天数循环
        su = su + i #天数相加
        if k > su: #如果k大于相加天数,说明前面的都是平方
            count = count + i*i #前面的总数
        else: #如果k小于相加天数,说明有多的天数,但是这部分不够平方
            count = count + (k-(su-i))*i 天数-(相加的天数减去i)就是多的几天,再乘以i就是多出来天数的金额
            break #执行到这里就结束了,直接跳出循环
    print(count)