0
点赞
收藏
分享

微信扫一扫

UVA714 抄书 Copying Books

Soy丶sauce 2022-02-03 阅读 17

知识点:二分

首先读题就看出来这个是最小化最大值的二分,从右向左的,然后要注意的是由于是求连续子列的和,所以有可能会出现溢出,需要在相应的地方用longlong,这个题的二分不难写,就是输出很麻烦,它要保证两个条件,一个是前面的子列尽可能小,但是还不能为空,至少要有一个,所以就想到了从后向前遍历,需要分类讨论,这个地方改了很多次终于改对了,

#include <bits/stdc++.h>

#define fi first
#define se second
#define pb push_back
#define mk make_pair
#define sz(x) ((int) (x).size())
#define all(x) (x).begin(), (x).end()

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pa;

const int N = 505;

int n, k, a[N];

bool check(ll x) {
    ll sum = 0;
    int cnt = 1;
    for (int i = 0; i < n; i++) {
        if (a[i] > x) return false;
        if (sum + a[i] <= x) sum += a[i];
        else { cnt++; sum = a[i]; }
    }
    return cnt <= k;
}

void solve(ll l, ll r) {
    while (l < r) {
        ll mid = (l + r) >> 1;
        if (check(mid)) r = mid;
        else l = mid + 1;
    }
    int t = k;
    ll sum = 0;
    vi v[N];
    for (int i = n - 1; i >= 0; i--) {
        if (sum + a[i] <= l) {
            if (t - 1 == i) {
                v[t--].pb(a[i]);
                sum = 0;
            } else {
                sum += a[i];
                v[t].pb(a[i]);
            }
        } else {
            v[--t].pb(a[i]);
            sum = a[i];
        }
    }
    for (int i = 1; i <= k; i++) {
        for (int j = sz(v[i]) - 1; j >= 0; j--) cout << v[i][j] << (i == k && !j ? "" : " ");
        if (i < k) cout << "/ ";
    }
    cout << endl;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        cin >> n >> k;
        for (int i = 0; i < n; i++) cin >> a[i];
        solve(0, 5000000000);
    }
    return 0;
}
举报

相关推荐

0 条评论