[CF1342D] Multiple Testcases

[CF1342D] Multiple Testcases - 构造

Description

给定 n 个整数，要放入若干个容器中，每个容器中，对于任意 i，大于等于 i 的数不能超过 ci 个。求最小所需容器数，构造方案。

Solution

对任意 i，计算 di 表示大于等于 i 的 a 有多少个，则 di 除以 ci 即可得到一堆限制，取最大的，就是组数

既然已经确定了有多少组，构造方案时，将所有数排序后轮流填入即可

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

#define int long long

signed main()
{
    ios::sync_with_stdio(false);

    int n, k;
    cin >> n >> k;
    vector<int> a(n + 2), c(k + 2);
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    for (int i = 1; i <= k; i++)
        cin >> c[i];

    vector<int> d(k + 2);
    for (int i = 1; i <= n; i++)
        d[a[i]]++;
    for (int i = k; i >= 1; i--)
        d[i] += d[i + 1];

    int ans = 0;
    for (int i = 1; i <= k; i++)
        ans = max(ans, (d[i] + c[i] - 1) / c[i]);

    vector<vector<int>> o(ans + 2);
    sort(&a[1], &a[n + 1]);
    for (int i = 1; i <= n; i++)
        o[i % ans + 1].push_back(a[i]);

    cout << ans << endl;
    for (int i = 1; i <= ans; i++)
    {
        cout << o[i].size() << " ";
        for (auto j : o[i])
            cout << j << " ";
        cout << endl;
    }
}