秉着能用Python就绝对不用C++的精神
Python秒了！

n=int(input());
s=0;
x=0;
m=1;
for i in range(1,n+1):
    s=s+1;
    if(s>m):
        m=m+1;
        s=1;
    x=x+m;
print(x);