Submission #1226865

#	Time	Username	Problem	Language	Result	Execution time	Memory
1226865		Aviansh	Cookies (JOI23_cookies)	C++20	7 / 100	48 ms	1156 KiB

cookies

#include <bits/stdc++.h>

using namespace std;

signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n;
    cin >> n;
    array<int,2> a[n];
    int sum = 0;
    for(int i = 0;i<n;i++){
        cin >> a[i][0];
        sum+=a[i][0];
        a[i][1]=i;
    }
    int m;
    cin >> m;
    int b[m];
    for(int &i : b){
        cin >> i;
    }
    assert(m==1);
    vector<vector<int>>v;
    while(sum>0){
        sort(a,a+n);
        v.push_back({});
        for(int i = n-1;i>=n-b[0];i--){
            v.back().push_back(a[i][1]);
            a[i][0]--;
        }
        sum-=b[0];
    }
    for(int i = 0;i<n;i++){
        if(a[i][0]!=0){
            cout << -1;
            return 0;
        }
    }
    cout << v.size() << "\n";
    for(vector<int>x : v){
        cout << x.size() << " ";
        for(int i : x){
            cout << i+1 << " ";
        }
        cout << "\n";
    }
    return 0;
}

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