Submission #866439

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
866439	2023-10-26T07:35:20 Z	SorahISA	Job Scheduling (IOI19_job)	C++17	19 / 100	157 ms	57676 KB

#include "job.h"
#include <bits/stdc++.h>
using namespace std;

using int64 = long long;

struct Job {
    int64 u, d, id;
    Job(int64 _u, int64 _d, int64 _id) : u(_u), d(_d), id(_id) {}
    bool operator < (const Job &rhs) const {
        return d * rhs.u < rhs.d * u;
    }
};

int64 scheduling_cost(vector<int> p, vector<int> u, vector<int> d) {
    int N = p.size();
    vector<vector<int>> ch(N);
    for (int i = 1; i < N; ++i) ch[p[i]].emplace_back(i);
    
    multiset<Job> jobs;
    for (int i = 0; i < N; ++i) jobs.emplace(u[i], d[i], i);
    
    int64 tim = 0, ans = 0;
    while (jobs.size()) {
        
        Job job = *begin(jobs);
        jobs.erase(begin(jobs));
        
        // cerr << "(" << job.u << ", " << job.d << ", " << job.id << ")\n";
        if (u[job.id] != job.u or d[job.id] != job.d) continue;
        // cerr << "(" << job.u << ", " << job.d << ", " << job.id << ")\n";
        
        if (p[job.id] == -1) {
            ans += job.u * (tim += job.d);
            for (int x : ch[job.id]) p[x] = -1;
            ch[job.id].clear(), ch[job.id].shrink_to_fit();
        }
        else {
            ans -= u[p[job.id]] * job.d;
            int64 new_u = u[p[job.id]] + job.u;
            int64 new_d = d[p[job.id]] + job.d;
            
            if (ch[p[job.id]].size() <= ch[job.id].size()) {
                for (int x : ch[p[job.id]]) {if (x != job.id) ch[job.id].emplace_back(x), p[x] = job.id;}
                ch[p[job.id]].clear(), ch[p[job.id]].shrink_to_fit();
                if (p[p[job.id]] != -1) *find(begin(ch[p[p[job.id]]]), end(ch[p[p[job.id]]]), p[job.id]) = job.id;
                p[job.id] = p[p[job.id]];
                jobs.emplace(u[job.id] = new_u, d[job.id] = new_d, job.id);
            }
            else {
                for (int x : ch[job.id]) ch[p[job.id]].emplace_back(x), p[x] = p[job.id];
                ch[job.id].clear(), ch[job.id].shrink_to_fit();
                jobs.emplace(u[p[job.id]] = new_u, d[p[job.id]] = new_d, p[job.id]);
            }
        }
    }
    
    return ans;
}

Subtask #1
0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	0 ms	348 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #2
7.0 / 7.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	125 ms	23132 KB	Output is correct
5	Correct	126 ms	22984 KB	Output is correct
6	Correct	122 ms	23016 KB	Output is correct
7	Correct	157 ms	23056 KB	Output is correct
8	Correct	144 ms	23132 KB	Output is correct
9	Correct	114 ms	23088 KB	Output is correct
10	Correct	118 ms	23076 KB	Output is correct
11	Correct	117 ms	22980 KB	Output is correct

Subtask #3
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	1 ms	348 KB	Output is correct
5	Correct	4 ms	1372 KB	Output is correct
6	Correct	118 ms	22984 KB	Output is correct
7	Correct	130 ms	23032 KB	Output is correct
8	Correct	117 ms	22900 KB	Output is correct
9	Correct	116 ms	23004 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	1 ms	604 KB	Output is correct
12	Correct	4 ms	1372 KB	Output is correct
13	Correct	4 ms	1372 KB	Output is correct
14	Correct	118 ms	22984 KB	Output is correct
15	Correct	127 ms	23100 KB	Output is correct
16	Correct	113 ms	23108 KB	Output is correct
17	Correct	141 ms	22920 KB	Output is correct
18	Correct	139 ms	23088 KB	Output is correct
19	Correct	135 ms	22988 KB	Output is correct
20	Correct	117 ms	23128 KB	Output is correct
21	Correct	118 ms	22980 KB	Output is correct
22	Correct	115 ms	22888 KB	Output is correct
23	Correct	116 ms	22996 KB	Output is correct
24	Correct	119 ms	22984 KB	Output is correct