洛谷-P2669 金币

题目链接

思路：

一个循环求出金币数即可。

代码：

#include<bits/stdc++.h>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=2e6+7;
const int M=2e4+5;
const double eps=1e-8;
const int mod=998244353;
const int inf=0x7fffffff;
const double pi=3.1415926;
using namespace std;
signed main()
{
    IOS;
    int K,N,ans=0;
    cin>>k;
    for(N=1;K-N>=0;K-=N++)
    {
        ans+=N*N;
    }
    cout<<ans+K*N<<endl;
    return 0;
}