Submission #1006070

#	Time	Username	Problem	Language	Result	Execution time	Memory
1006070		MilosMilutinovic	Cookies (JOI23_cookies)	C++14	6 / 100	4 ms	6604 KiB

cookies

#include <bits/stdc++.h>

using namespace std;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n;
  cin >> n;
  vector<int> a(n);
  for (int i = 0; i < n; i++) {
    cin >> a[i];
  }
  int m;
  cin >> m;
  vector<int> b(m);
  for (int i = 0; i < m; i++) {
    cin >> b[i];
  }
  const int MAX = 505;
  vector<int> cnt(MAX);
  for (int i = 0; i < n; i++) {
    cnt[a[i]] += 1;
  }
  for (int i = MAX - 1; i > 0; i--) {
    cnt[i - 1] += cnt[i];
  }
  int mx = *max_element(a.begin(), a.end());
  vector<vector<int>> dp(MAX, vector<int>(MAX, MAX));
  vector<vector<pair<int, int>>> prv(MAX, vector<pair<int, int>>(MAX, {-1, -1}));
  auto Update = [&](int i, int j, int x, int y, int f) {
    if (dp[i][j] > dp[x][y] + f) {
      dp[i][j] = dp[x][y] + f;
      prv[i][j] = {x, y};
    }
  };
  dp[mx][0] = 0;
  for (int c = mx; c >= 1; c--) {
    for (int x = 0; x <= cnt[c]; x++) {
      if (dp[c][x] == MAX) {
        continue;
      }
      for (int i = 0; i < m; i++) {
        int limit = (c == 1 ? cnt[c] - x : cnt[c - 1]);
        if (b[i] > limit) {
          continue;
        }
        int y = x + b[i];
        if (y <= cnt[c]) {
          Update(c, y, c, x, 1);
        } else {
          Update(c - 1, y - cnt[c], c, x, 1);
        }
      }
    }
    Update(c - 1, 0, c, cnt[c], 0);
  }
  if (dp[1][cnt[1]] == MAX) { 
    cout << -1 << '\n';
    return 0; 
  }
  int x = 1, y = cnt[1];
  vector<int> bv;
  while (x >= 1 && x <= mx) {
    int pv;
    if (prv[x][y].first == -1) {
      pv = 0;
    } else if (prv[x][y].first == x) {
      pv = prv[x][y].second;
    } else {
      pv = prv[x][y].second - cnt[x + 1];
    }
    if (y != pv) {
      bv.push_back(y - pv);
    }
    pair<int, int> p = prv[x][y];
    x = p.first;
    y = p.second;
  }
  set<pair<int, int>> st;
  for (int i = 0; i < n; i++) {
    st.emplace(a[i], i);
  }
  vector<vector<int>> res;
  for (int v : bv) {
    res.push_back({});
    for (int iter = 0; iter < v; iter++) {
      auto it = prev(st.end());
      int i = it->second;
      st.erase(it);
      res.back().push_back(i);
    }
    for (int i : res.back()) {
      a[i] -= 1;
      if (a[i] > 0) {
        st.emplace(a[i], i);
      }
    }
  }
  cout << (int) res.size() << '\n';
  for (int i = 0; i < (int) res.size(); i++) {
    cout << (int) res[i].size() << " ";
    for (int j = 0; j < (int) res[i].size(); j++) {
      cout << res[i][j] + 1 << " ";
    }
    cout << '\n';
  }
  return 0;
}

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