很疑惑，样例是对的，也尝试了其他数字，但是提交就全wa

include <stdio.h>

int main()
{
	int k,s;
	scanf("%d",&k);
	int d = 0;
	int i = 0;
    while(k>d)//算出k是几组数的和
    {
    	i++;
    	d += i;
	}
	for(int j = 1;j<=i;j++)//计算整组的金币数量（假设全部都是整组）
	{
		s += j * j;
	}
	if((d-k)!=0)
	{
	    s = s-i*(d-k);	//减去多余的金币数量
	} 
	printf("%d",s);
   return 0;
}