Submission #305726

# Submission time Handle Problem Language Result Execution time Memory
305726 2020-09-23T21:50:35 Z aZvezda Job Scheduling (CEOI12_jobs) C++14
100 / 100
876 ms 21076 KB
#include <bits/stdc++.h>
using namespace std;
//#pragma GCC optimize ("O3")
//#pragma GCC target ("sse4")
#define endl "\n"
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
template<class T, class T2> inline bool chkmax(T &x, const T2 &y) { return x < y ? x = y, 1 : 0; }
template<class T, class T2> inline bool chkmin(T &x, const T2 &y) { return x > y ? x = y, 1 : 0; }
const ll mod = 1e9 + 7;
template<class T> inline void fix(T &x) {if(x >= mod | x <= -mod) {x %= mod;} if(x < 0) {x += mod;}}
#define out(x) cout << __LINE__ << ": " << (#x) << " = " << (x) << endl

const int MAX_N = 1e6 + 10;
pair<int, int> arr[MAX_N];
vector<int> p;
int n, d, m;

bool eval(int x) {
    priority_queue<int> pq;
    int ind = 0;
    for(int i = 1; i <= m; i ++) {
        while(ind < n && arr[ind].first == i) {
            pq.push(-arr[ind].first - d);
            ind ++;
        }
        for(int j = 0; j < x; j ++) {
            if(pq.empty()) {break;}
            pq.pop();
        }
        if(!pq.empty() && -pq.top() <= i) {
            return false;
        }
    }
    return true;
}

void print(int x) {
    priority_queue<pair<int, int> > pq;
    int ind = 0;
    for(int i = 1; i <= m; i ++) {
        while(ind < n && arr[ind].first == i) {
            pq.push({-arr[ind].first - d, arr[ind].second + 1});
            ind ++;
        }
        for(int j = 0; j < x; j ++) {
            if(pq.empty()) {break;}
            cout << pq.top().second << " ";
            pq.pop();
        }
        cout << 0 << endl;
    }
}

signed main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    cin >> m >> d >> n;
    for(int i = 0; i < n; i ++) {
        cin >> arr[i].first;
        arr[i].second = i;
        p.push_back(arr[i].first);
        p.push_back(arr[i].first + d);
    }
    sort(arr, arr + n);
    int l = 0, r = n + 1;
    while(l < r - 1) {
        int m = (l + r) / 2ll;
        if(eval(m)) {
            r = m;
        } else {
            l = m;
        }
    }
    cout << r << endl;
    print(r);
    return 0;

}

/*
8 2 12
1 2 4 2 1 3 5 6 2 3 6 4
*/
# Verdict Execution time Memory Grader output
1 Correct 103 ms 3444 KB Output is correct
2 Correct 110 ms 3572 KB Output is correct
3 Correct 103 ms 3488 KB Output is correct
4 Correct 106 ms 3592 KB Output is correct
5 Correct 109 ms 3496 KB Output is correct
6 Correct 104 ms 3448 KB Output is correct
7 Correct 104 ms 3488 KB Output is correct
8 Correct 106 ms 3480 KB Output is correct
9 Correct 105 ms 2928 KB Output is correct
10 Correct 117 ms 2928 KB Output is correct
11 Correct 72 ms 2672 KB Output is correct
12 Correct 234 ms 5224 KB Output is correct
13 Correct 271 ms 7136 KB Output is correct
14 Correct 407 ms 10208 KB Output is correct
15 Correct 359 ms 11748 KB Output is correct
16 Correct 397 ms 14032 KB Output is correct
17 Correct 578 ms 16340 KB Output is correct
18 Correct 876 ms 18772 KB Output is correct
19 Correct 874 ms 21076 KB Output is correct
20 Correct 581 ms 16296 KB Output is correct