#include <math.h>
#include <stdio.h>

int main() {
    int n,sum=0,i=1,j=0,op;
    scanf("%d",&n);
    op=n-1;
    do
    {   i++;
        op-=i;
    }while(op>0);
    if(op) op+=i;
    for(j=1;j<=i;j++,n-=j)
    {
        if((n&&j!=i)||n>=0) sum+=pow(j,2);
        else if(n&&j==i) sum+=(op*i);
    }
    printf("%d\n",sum);
}

给第一个学生（计算机）将数学题