oj.uz

Submission #1025424

# Submission time Handle Problem Language Result Execution time Memory
1025424 2024-07-17T03:06:34 Z shiomusubi496 Toy Train (IOI17_train) C++17
38 / 100
 231 ms 262144 KB 
#include "train.h"

#include <bits/stdc++.h>

#define rep(i, n) for (int i = 0; i < (int)(n); ++i)
#define rep2(i, a, b) for (int i = (int)(a); i < (int)(b); ++i)
#define rrep(i, n) for (int i = (int)(n) - 1; i >= 0; --i)
#define rrep2(i, a, b) for (int i = (int)(b) - 1; i >= (int)(a); --i)
#define all(v) begin(v), end(v)

using namespace std;

template<class T, class U> bool chmin(T& a, const U& b) { return a > b ? a = b, true : false; }
template<class T, class U> bool chmax(T& a, const U& b) { return a < b ? a = b, true : false; }

using ll = long long;

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
    int n = a.size(), m = u.size();
    vector<vector<int>> g(n), rg(n);
    rep (i, m) {
        g[u[i]].push_back(v[i]);
        rg[v[i]].push_back(u[i]);
    }
    auto calc = [&](vector<int> r, bool f) -> vector<int> {
        queue<int> que;
        rep (i, n) {
            if (r[i]) que.push(i);
        }
        vector<int> deg(n);
        rep (i, n) deg[i] = g[i].size();
        while (!que.empty()) {
            int i = que.front(); que.pop();
            for (int j : rg[i]) {
                if (r[j]) continue;
                if (a[j] == f) {
                    r[j] = true;
                }
                else {
                    if ((--deg[j]) == 0) {
                        r[j] = true;
                    }
                }
                if (r[j]) que.push(j);
            }
        }
        return r;
    };
    auto r2 = calc(r, 1);
    if (count(all(r2), 1) == n) return r2;
    rep (i, n) r2[i] ^= 1;
    auto r3 = calc(r2, 0);
    vector<int> idx(n, -1);
    int cnt = 0;
    rep (i, n) {
        if (!r3[i]) {
            idx[i] = cnt++;
        }
    }
    vector<int> a2(cnt), r4(cnt);
    rep (i, n) {
        if (idx[i] == -1) continue;
        a2[idx[i]] = a[i];
        r4[idx[i]] = r[i];
    }
    vector<int> u2, v2;
    rep (i, m) {
        if (idx[u[i]] == -1 || idx[v[i]] == -1) continue;
        u2.push_back(idx[u[i]]);
        v2.push_back(idx[v[i]]);
    }
    auto res = who_wins(a2, r4, u2, v2);
    vector<int> ans(n);
    rep (i, n) {
        if (idx[i] == -1) ans[i] = 0;
        else ans[i] = res[idx[i]];
    }
    return ans;
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 2652 KB Output is correct
2 Correct 6 ms 4436 KB Output is correct
3 Correct 6 ms 3676 KB Output is correct
4 Correct 5 ms 3516 KB Output is correct
5 Correct 5 ms 2908 KB Output is correct
6 Correct 5 ms 2908 KB Output is correct
7 Correct 3 ms 1380 KB Output is correct
8 Correct 7 ms 4760 KB Output is correct
9 Correct 5 ms 2396 KB Output is correct
10 Correct 4 ms 1884 KB Output is correct
11 Correct 3 ms 1628 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 1 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 1 ms 344 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 0 ms 348 KB Output is correct
16 Correct 0 ms 348 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 1 ms 348 KB Output is correct
19 Correct 1 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Runtime error 231 ms 262144 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #4
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 2096 KB Output is correct
2 Correct 6 ms 1884 KB Output is correct
3 Correct 5 ms 1628 KB Output is correct
4 Correct 6 ms 1700 KB Output is correct
5 Correct 6 ms 2140 KB Output is correct
6 Correct 6 ms 2472 KB Output is correct
7 Correct 8 ms 2352 KB Output is correct
8 Correct 5 ms 1628 KB Output is correct
9 Correct 6 ms 2360 KB Output is correct
10 Correct 5 ms 1628 KB Output is correct
11 Correct 5 ms 1628 KB Output is correct
12 Correct 5 ms 1628 KB Output is correct
13 Correct 6 ms 2136 KB Output is correct
14 Correct 7 ms 2080 KB Output is correct

Subtask #5
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 1884 KB Output is correct
2 Correct 5 ms 1628 KB Output is correct
3 Correct 8 ms 2468 KB Output is correct
4 Correct 6 ms 1488 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 3 ms 1116 KB Output is correct
7 Correct 3 ms 1372 KB Output is correct
8 Correct 3 ms 1248 KB Output is correct
9 Correct 3 ms 1372 KB Output is correct
10 Correct 1 ms 604 KB Output is correct
11 Correct 4 ms 1372 KB Output is correct

Subtask #6
0 / 51.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 2652 KB Output is correct
2 Correct 6 ms 4436 KB Output is correct
3 Correct 6 ms 3676 KB Output is correct
4 Correct 5 ms 3516 KB Output is correct
5 Correct 5 ms 2908 KB Output is correct
6 Correct 5 ms 2908 KB Output is correct
7 Correct 3 ms 1380 KB Output is correct
8 Correct 7 ms 4760 KB Output is correct
9 Correct 5 ms 2396 KB Output is correct
10 Correct 4 ms 1884 KB Output is correct
11 Correct 3 ms 1628 KB Output is correct
12 Correct 1 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 344 KB Output is correct
15 Correct 0 ms 348 KB Output is correct
16 Correct 0 ms 348 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 1 ms 344 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 0 ms 348 KB Output is correct
24 Correct 0 ms 348 KB Output is correct
25 Correct 0 ms 348 KB Output is correct
26 Correct 0 ms 348 KB Output is correct
27 Correct 0 ms 348 KB Output is correct
28 Correct 0 ms 348 KB Output is correct
29 Correct 1 ms 348 KB Output is correct
30 Correct 1 ms 348 KB Output is correct
31 Correct 0 ms 348 KB Output is correct
32 Runtime error 231 ms 262144 KB Execution killed with signal 9
33 Halted 0 ms 0 KB -