提交时间:2024-02-10 15:05:02

运行 ID: 40414

#include<iostream> using namespace std; int main() { int k,ans=0; cin>>k; if(k==1){ cout<<"1"; } else{ for(int i=0;i<k;i++){ if(i*(i+1)/2<=k&&k<(i+1)*(i+2)/2){ for(int j=0;j<=i;j++){ ans+=j*j; } cout<<ans+(k-i*(i+1)/2)*(i+1); break; } } } }