Submission #970684

#	Time	Username	Problem	Language	Result	Execution time	Memory
970684		huutuan	Cookies (JOI23_cookies)	C++14	0 / 100	1 ms	604 KiB

cookies

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

#ifdef sus
const int N=100;
#else
const int N=3010;
#endif
vector<vector<int>> f[N];
int n, m, a[N], b[N], pf[N], c[N], _a[N];
 
int32_t main(){
   ios_base::sync_with_stdio(false);
   cin.tie(nullptr);
   cin >> n;
   for (int i=1; i<=n; ++i) cin >> _a[i];
   cin >> m;
   for (int i=1; i<=m; ++i) cin >> b[i], c[b[i]]=1;
   for (int i=1; i<=n; ++i) a[i]=_a[i];
   int sum=accumulate(a+1, a+n+1, 0);
   sort(a+1, a+n+1, greater<int>());
   for (int i=1, j=n; i<=sum; ++i){
      pf[i]=pf[i-1]+j;
      while (j && a[j]==i) --j;
   }
   f[0].resize(sum+1, vector<int>(sum+1));
   for (int i=1; i<=sum; ++i){
      f[i].resize(sum/i+1, vector<int>(sum+1));
   }
   for (int i=0; i<=sum; ++i) f[0][i][0]=1;
   vector<int> sz;
   for (int i=1; i<=sum; ++i){
      for (int j=sum/i; j>=0; --j){
         for (int k=i*j; k<=pf[i]; ++k){
            f[i][j][k]|=f[i][j+1][k];
            if (c[j]) f[i][j][k]|=f[i-1][j][k-j];
            if (k==sum && f[i][j][k]){
               while (i){
                  if (c[j] && k>=j && f[i-1][j][k-j]){
                     sz.push_back(j);
                     --i;
                     k-=j;
                  }else ++j;
               }
               goto done;
            }
         }
      }
   }
   cout << -1 << '\n';
   return 0;
done:
   reverse(sz.begin(), sz.end());
   priority_queue<pair<int, int>> pq;
   vector<vector<int>> ans;
   for (int i=1; i<=n; ++i) a[i]=_a[i];
   for (int i=1; i<=n; ++i) pq.emplace(a[i], i);
   for (int i:sz){
      ans.emplace_back();
      for (int j=0; j<i; ++j){
         if (pq.empty()){
            cout << -1 << '\n';
            return 0;
         }
         ans.back().push_back(pq.top().second);
         pq.pop();
      }
      for (int j:ans.back()){
         --a[j];
         if (a[j]) pq.emplace(a[j], j);
      }
   }
   cout << ans.size() << '\n';
   for (auto &i:ans){
      cout << i.size();
      for (int j:i) cout << ' ' << j;
      cout << '\n';
   }
   return 0;
}

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