Codeforces 1157D N Problems During K Days

題目傳送門

#include<iostream>
#include<cmath>
using namespace std;
long long a[110000];
long long s(long long m)
{
    return m*(m+1)/2;
}
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=k;i++)//一次循環确定一個數
    {
        long long ans=max((int)(a[i-1]+1),(int)ceil((double)n/(double)(pow(2.0,k+1-i)-1)));//一定要用上取整。
        if(s(ans-i+k)-s(ans-1)>n)//這句話的意思是，确定了前i-1個數，然後讓第i個數一直到第k個數是兩兩之間是以1為公差的數，判斷第i個數開始加是否會超過n，如果超過了就不能完成。
        {
            cout<<"NO"<<endl;
            return 0;
        }
        a[i]=ans;
        n-=ans;//每确定一個數，将其在總數上删去。
    }
    cout<<"YES"<<endl;
    for(int i=1;i<=k;i++)
    {
        cout<<a[i]<<" ";
    }
    return 0;
}