牛客周赛 Round 59 D.构造mex

题目链接：题目

大意：

给定总和s，长度n，和最小未出现整数k，构造一个可行数组。

思路：

构造题一般都是思考最容易构造的情况，再具体分情况讨论。要满足k的条件，那么就直接在数组中插入0, 1, 2, …, k就行了，如果无法完整插入，那么就不行。如果插入的总和已经超过s，那么也不行。如果能插入并且还有剩，那就再插入剩下的，再补充0。特殊情况，如果剩下的恰好等于k，又只剩下一个位置，那么也不行，如果剩至少两个位置，就可以分成两半，1和k-1，如果剩下的就是1，并且k=1，那么还是分不了。还有一种特殊情况，k=0，那么就先放1，最后一个再补剩下的，如果光放1已经超出s了，那么也不行。

代码：

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define MOD 1000000007
#define fi first
#define se second
#define pii pair<int,int>
#define vec vector

void solve(){
    int s, n, k;
    cin >> s >> n >> k;
    int a = (k - 1) * k / 2, b = (n - 1) * n / 2;
    if(k == 0){
        if(s < n){
            cout << "NO" << '\n';
            return;
        }
        cout << "YES" << '\n';
        for(int i = 0;i < n - 1;i++){
            cout << 1 << ' ';
        }
        cout << s - (n - 1) << '\n';
        return;
    }
    if(k > n){
        cout << "NO" << '\n';
        return;
    }
    if(k == n){
        if(s == a){
            cout << "YES" << '\n';
            for(int i = 0;i < n;i++){
                cout << i << ' ';
            }
            return;
        }
        cout << "NO" << '\n';
        return;
    }
    if(a > s){
        cout << "NO" << '\n';
        return;
    }
    if(k < n){
        if(k == n - 1){
            if(s - a == k){
                cout << "NO" << '\n';
                return;
            }
            cout << "YES" << '\n';
            for(int i = 0;i < k;i++){
                cout << i << ' ';
            }
            cout << s - a << '\n';
            return;
        }
        if(s - a == k){
            if(k == 1){
                cout << "NO" << '\n';
                return;
            }
            cout << "YES" << '\n';
            for(int i = 0;i < k;i++){
                cout << i << ' ';
            }
            for(int i = 1;i <= n - k - 2;i++){
                cout << 0 << ' ';
            }
            cout << 1 << ' ' << k - 1 << '\n';
            return;
        }
        if(s - a < 0){
            cout << "NO" << '\n';
            return;
        }
        cout << "YES" << '\n';
        for(int i = 0;i < k;i++){
            cout << i << ' ';
        }
        for(int i = 1;i <= n - k - 1;i++){
            cout << 0 << ' ';
        }
        cout << s - a << '\n';
        return;
    }
}

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int t=1;
    cin >> t;
    while(t--){
        solve();
    }
    return 0;
}