Submission #1025417

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
1025417	2024-07-17T02:38:57 Z	shiomusubi496	Toy Train (IOI17_train)	C++17	0 / 100	8 ms	2652 KB

train

#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();
    if (n == 0) return {};
    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;
                        que.push(j);
                    }
                }
                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);
    cout << flush;
    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
0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	5 ms	2648 KB	3rd lines differ - on the 42nd token, expected: '0', found: '1'
2	Halted	0 ms	0 KB	-

Subtask #2
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	344 KB	Output is correct
2	Correct	1 ms	344 KB	Output is correct
3	Correct	1 ms	344 KB	Output is correct
4	Correct	1 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Incorrect	0 ms	344 KB	3rd lines differ - on the 1st token, expected: '0', found: '1'
7	Halted	0 ms	0 KB	-

Subtask #3
0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4 ms	1624 KB	Output is correct
2	Correct	5 ms	1628 KB	Output is correct
3	Correct	5 ms	1628 KB	Output is correct
4	Correct	8 ms	2652 KB	Output is correct
5	Incorrect	6 ms	1628 KB	3rd lines differ - on the 1st token, expected: '1', found: '0'
6	Halted	0 ms	0 KB	-