Submission #528697

#TimeUsernameProblemLanguageResultExecution timeMemory
528697AanjaneyJob Scheduling (CEOI12_jobs)C++17
80 / 100
450 ms45360 KiB
#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define MOD 1000000007
#define MODA 998244353
#define pb push_back
#define mp make_pair
#define sortv(v) sort(v.begin(), v.end())
#define sorta(A, N) sort(A, A + N)
#define debug(x) cerr << #x << " is " << x;
#define rep(i, a, N) for (ll i = a; i < N; i++)
#define f first
#define s second
#define uniq(v)                                       \
    {                                                 \
        sort(v.begin(), v.end());                     \
        v.erase(unique(v.begin(), v.end()), v.end()); \
    }
#define speed                         \
    ios_base::sync_with_stdio(false); \
    cin.tie(NULL);                    \
    cout.tie(NULL);
using namespace std;
ll N, D, M;
pair<bool, vector<vector<ll> > > check(const vector<pair<ll, ll> > &jobs, ll machineCount)
{
    vector<vector<ll> > schedule(N);
    ll reqNum = 0;
    rep(day, 1, N + 1)
    {
        rep(j, 0, machineCount)
        {
            if (jobs[reqNum].f > day)
                break;
            if (jobs[reqNum].f + D >= day)
                schedule[day - 1].pb(jobs[reqNum++].s);
            else
                return mp(false, schedule);

            if (reqNum == M)
                return mp(true, schedule);
        }
    }
    return mp(false, schedule);
}
void solve(ll tcase)
{
    cin >> N >> D >> M;
    vector<pair<ll, ll> > jobs(M);
    rep(i, 0, M)
    {
        ll day;
        cin >> day;
        jobs[i] = mp(day, i + 1);
    }
    sortv(jobs);
    vector<vector<ll> > result;
    ll lo = 1, hi = M + 1;
    while (lo < hi)
    {
        ll machineNum = (lo + hi) / 2;
        pair<bool, vector<vector<ll> > > curResult = check(jobs, machineNum);
        if (curResult.f)
        {
            hi = machineNum;
            result = curResult.s;
        }
        else
            lo = machineNum + 1;
    }

    cout << lo << "\n";
    rep(i, 0, N)
    {
        for (auto j : result[i])
            cout << j << ' ';
        cout << 0 << "\n";
    }
}
int main()
{

    speed;
    ll t = 1;
    rep(i, 1, t + 1)
        solve(i);
}
#Verdict Execution timeMemoryGrader output
Fetching results...