#include <bits/stdc++.h>
using namespace std;
int main()
{
	int n,i,answer=0;
	cin>>n;
    for (i=1;i<=n;i++)
	{
        if (n-i>=0)
        {
		   n-=i;
		   answer+=i*i;
		}
	}
	answer+=(i+1)*n;
    cout<<answer;
	return 0;
}

本题只得了40分，不知道是哪里出了问题。麻烦指出错误，并给出正确改正方案。谢谢！