Bonetrousle HackerRank 数学 + 思维题

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
const int maxn = 1e5 + 20;
LL a[maxn];
void work () {
//    ios::sync_with_stdio(false);
    LL n, k, b;
//    scanf ("%lld%lld%lld", &n, &k, &b);
    cin >> n >> k >> b;
    //cout << n << " " << k << " " << b << endl;
    LL mi = 0, mx = 0;
    for (int i = 1; i <= b; ++i) {
        mi += i;
    }
    for (LL i = k; i >= k - b + 1; --i) {
        mx += i;
        //if (mx < 0) while (1);
        if (mx >= n) break; // ok
    }
    //cout << mi << " " << mx << endl;
    if (mx < n) {
        printf ("-1\n");
        return ;
    }
    if (mi > n) {
        printf ("-1\n");
        return ;
    }
    LL now = mi;
    int to = b;
    LL get = k;
    for (int i = 1; i <= b; ++i) a[i] = i;
    while (true) {
        now = now - a[to] + get;
        a[to] = get;
        if (now >= n) {
            LL cut = now - n;
            a[to] -= cut;
            for (int i = 1; i <= b - 1; ++i) {
                printf ("%lld ", a[i]);
            }
            printf ("%lld\n", a[b]);
            break;
        }
        to--;
        get--;
    }
}

int main ()
{
#ifdef LOCAL
    freopen("data.txt","r",stdin);
#endif
    int t;
    cin >> t;
//    cout << t << endl;
    while (t--) work ();
    return 0;
}