Submission #679819

#TimeUsernameProblemLanguageResultExecution timeMemory
679819peijarJob Scheduling (CEOI12_jobs)C++17
100 / 100
403 ms21560 KiB
#include <bits/stdc++.h>
#define int long long
using namespace std;

namespace std {
template <typename T> ostream &operator<<(ostream &out, const vector<T> &vec) {
  out << "[";
  for (int i = 0; i < (int)vec.size(); ++i) {
    out << vec[i];
    if (i + 1 < (int)vec.size())
      out << ", ";
  }
  return out << "]";
}
} // namespace std

void dbg_out() { cout << endl; }
template <typename Head, typename... Tail> void dbg_out(Head H, Tail... T) {
  cout << ' ' << H;
  dbg_out(T...);
}

#ifdef DEBUG
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif

signed main(void) {
  ios_base::sync_with_stdio(false);
  cin.tie(0);

  int nbJours, nbTravaux, deadline;
  cin >> nbJours >> deadline >> nbTravaux;

  vector<int> debut(nbTravaux);
  for (int &x : debut) {
    cin >> x;
    --x;
  }
  vector<int> order(nbTravaux);
  iota(order.begin(), order.end(), 0);
  sort(order.begin(), order.end(),
       [&](int i, int j) { return debut[i] < debut[j]; });

  auto check = [&](int nbMachines) -> bool {
    queue<int> dispo;
    for (int i = 0; i < nbJours; ++i)
      dispo.push(i);
    vector<int> surTemps(nbJours);
    for (int i : order) {
      int t = debut[i];
      while (!dispo.empty() and dispo.front() < t)
        dispo.pop();
      if (dispo.empty() or dispo.front() > t + deadline)
        return false;

      int tt = dispo.front();
      if (tt > t + deadline)
        return false;
      surTemps[tt]++;
      if (surTemps[tt] == nbMachines)
        dispo.pop();
    }
    return true;
  };

  int lo = 1, up = nbTravaux;
  while (lo < up) {
    int mid = (lo + up) / 2;
    if (check(mid))
      up = mid;
    else
      lo = mid + 1;
  }
  cout << lo << endl;
  int curJour = 0;
  vector<int> onDay;
  vector<int> surTemps(nbJours);
  for (int i : order) {
    int t = debut[i];
    while (curJour < t or surTemps[curJour] == lo) {
      for (int x : onDay)
        cout << x << ' ';
      cout << 0 << '\n';
      onDay.clear();
      curJour++;
    }
    assert(curJour < nbJours and curJour <= t + deadline);
    onDay.push_back(i + 1);
    surTemps[curJour]++;
  }
  while (curJour < nbJours) {
    for (int x : onDay)
      cout << x << ' ';
    cout << 0 << '\n';
    onDay.clear();
    curJour++;
  }
}
#Verdict Execution timeMemoryGrader output
Fetching results...