洛谷上一道入门题，我自己试着答案都对但提交后一个测试点都没过

这是我的代码

#include <bits/stdc++.h>
using namespace std;
int main(){
int a,b,i,j=0,m,n;
scanf("%d",&a);
for(i=1;j<=a;i++){
    
if(j+i<=a){    j=i+j;
    m+=i*i;}else{m+=(a-j)*i;break;
    }
}
    cout<<m;
}

#include <iostream>
#include <cmath>

using namespace std;

int main()
{
    int k, n, c = 0, s = 0;
    std::cin >> k;
    n = (int)std::sqrt(2 * k); // (n+1)n/2 <= k
    for (int i = 1; i <= n; i++)
    {
        c += i;
        s += i * i;
    }
    s += (k - c) * (n + 1);
    std::cout << s;
    return 0;
}