Submission #807227

#	Time	Username	Problem	Language	Result	Execution time	Memory
807227		someone	Cookies (JOI23_cookies)	C++14	6 / 100	23 ms	29604 KiB

cookies

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

const int N = 1.5e4 + 42;

vector<int> boite;
vector<bitset<N>> dp[N];
int n, m, a[N], sz[N], prefix[N];

void remonter(int i, int j, int sum) {
    if(sum == 0)
        return;
    if(dp[i-1][j][sum - sz[j]]) {
        boite.push_back(-sz[j]);
        remonter(i-1, j, sum - sz[j]);
    } else {
        remonter(i, j+1, sum);
    }
}

void printSol() {
    vector<int> id[N];
    for(int i = 0; i < n; i++)
        for(int j = 0; j < a[i]; j++)
            id[j].push_back(i+1);
    queue<int> ids;
    for(int i = 0; i < N; i++)
        for(int j : id[i])
            ids.push(j);
    sort(boite.begin(), boite.end());
    for(int b : boite) {
        b = -b;
        cout << b << ' ';
        for(int i = 0; i < b; i++) {
            cout << ids.front() << ' ';
            ids.pop();
        }
        cout << '\n';
    }
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    cin >> n;
    for(int i = 0; i < n; i++) {
        cin >> a[i];
        for(int j = 0; j < N; j++)
            prefix[j] += min(j, a[i]);
    }
    cin >> m;
    dp[0].resize(m);
    dp[0][m-1] = true;
    for(int i = 0; i < m; i++)
        cin >> sz[i];
    int sum = prefix[N-1];
    for(int i = 0; i <= sum; i++) {
        for(int j = m-1; j >= 0; j--)
            if(i * sz[j] <= sum) {
                dp[i+1].resize(j+1);
                j = 0;
            }
        for(int j = m-1; j >= 0; j--)
            if(i * sz[j] <= sum) {
                dp[i][j] <<= (N-prefix[i]-1);
                dp[i][j] >>= (N-prefix[i]-1);
                if(dp[i][j][sum]) {
                    remonter(i, j, sum);
                    cout << i << '\n';
                    printSol();
                    return 0;
                }
                if(j > 0)
                    dp[i][j-1] |= dp[i][j];
                if((i+1) * sz[j] <= sum)
                    dp[i+1][j] |= (dp[i][j] << sz[j]);
            }
    }
    cout << -1 << '\n';
}

• Subtask 1: 6 / 6
• Subtask 2: 0 / 7
• Subtask 3: 0 / 12
• Subtask 4: 0 / 45
• Subtask 5: 0 / 15
• Subtask 6: 0 / 15