제출 #531227

#제출 시각아이디문제언어결과실행 시간메모리
531227ceoi_mikeJob Scheduling (CEOI12_jobs)C++11
100 / 100
555 ms20072 KiB
#include <bits/stdc++.h>
using namespace std;

typedef pair<int,int> pi;
typedef vector<int> vi;
typedef vector<pi> vpi;
int N, M, D;

bool do_schedule(vector<pi>& jobs, int n_mach, vector<vi>& schedule) {
    // job id. how do we look up day. maybe job_idx
    // we can look up day and job id
    schedule.clear();
    for (int i = 0; i <= N; i++) {
        schedule.push_back(vi());
    }
    int job_idx = 0;
    for (int day = 1; day <= N; day++) {
        int mach_left = n_mach;
        while (job_idx < M && jobs[job_idx].first <= day && mach_left > 0) {
            schedule[day].push_back(job_idx);
            job_idx++;
            mach_left--;
        }
        // check if any days are past deadline
        for (int i = 0; i < (int)schedule[day].size(); i++) {
            int scheduled_day = jobs[schedule[day][i]].first;
            int last_day_good = scheduled_day + D;
            if (day > last_day_good) {
                return false;
            }
        }
    }
    return true;
}

void show_schedule(vector<pi>& jobs, vector<vi>& schedule) {
    for (int day = 1; day <= N; day++) {
        for (int idx : schedule[day]) {
            cout << jobs[idx].second << " ";
        }
        cout << "0" << endl;
    }
}

int main()
{
    cin >> N >> D >> M;
    vector<pi> jobs(M);
    for (int i = 0; i < M; i++) {
        int sched;
        cin >> sched;
        jobs[i] = pi(sched, i+1);
    }
    sort(begin(jobs), end(jobs));
    // bool do_schedule(vector<pi>& jobs, int n_mach, vector<vi>& schedule) 
    vector<vi> schedule;
    // do_schedule(jobs, 2, schedule);
    // show_schedule(jobs, schedule);

    int lo = 1;
    int hi = M+1;
    // lo is one above false. hi is true
    while (lo < hi) {
        int mid = lo + (hi-lo)/2;
        if (do_schedule(jobs, mid, schedule)) {
            hi = mid;
        } else {
            lo = mid+1;
        }
    }
    do_schedule(jobs, hi, schedule);
    cout << hi << endl;
    show_schedule(jobs, schedule);
}
#Verdict Execution timeMemoryGrader output
Fetching results...