很疑惑，样例是对的，也尝试了其他数字，但是提交就全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;
}