Submission #1147441

#	Time	Username	Problem	Language	Result	Execution time	Memory
1147441		Hamed_Ghaffari	Cookies (JOI23_cookies)	C++20	7 / 100	3 ms	1152 KiB

cookies

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

using pii = pair<int, int>;

#define X first
#define Y second
#define SZ(x) int(x.size())
#define pb push_back

const int MXN = 500;

int n, m, A[MXN];

int32_t main() {
  cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0);
  cin >> n;
  assert(n<=500);
  for(int i=0; i<n; i++) cin >> A[i];
  cin >> m;
  if(m==1) {
    int b;
    cin >> b;
    vector<vector<int>> ans;
    priority_queue<pii> pq;
    for(int i=0; i<n; i++) pq.push(pii(A[i],i));
    while(SZ(pq)>=b) {
      vector<int> vec;
      for(int i=0; i<b; i++) {
        vec.pb(pq.top().Y); pq.pop();
      }
      ans.pb(vec);
      for(int i : vec) if(A[i]>1) pq.push(pii(--A[i], i));
    }
    if(!pq.empty()) {
      cout << "-1\n";
      return 0;
    }
    cout << SZ(ans) << '\n';
    for(auto i : ans) {
      cout << SZ(i) << ' ';
      for(auto j : i) cout << j+1 << ' ';
      cout << '\n';
    }
  }
  else assert(0);
  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