为啥会WA两个点

#include<cstdio>
#include<cmath>
using namespace std;
int k,d;
int main(){
    scanf("%d",&k);
    d=sqrt(2*k);
    printf("%d",d*(d+1)*(2*d+1)/6+(k-(1+d)*d/2)*(d+1));
}