我输入输出样例一模一样,但是全WA,我下载了第一个测试点输入6输出14跟我的一模一样,我不理解我哪里错了呀有没有大佬帮帮
#include<bits/stdc++.h>
using namespace std;
int main(){
int b,k,i,s,a,sum;
cin>>k;
s=0;
i=0;
while(s<k){
i++;
s+=i;
}
for(int n=1;n<=i;n++){
a=n;
sum+=pow(a,2);
}
if(s!=k){
b=s-k;
sum=sum-b*i;
}
cout<<sum<<endl;
return 0;
}