Submission #797694

#	Time	Username	Problem	Language	Result	Execution time	Memory
797694		PixelCat	Toy Train (IOI17_train)	C++14	0 / 100	7 ms	1636 KiB

train

#ifdef NYAOWO
#include "grader.cpp"
#endif

#include "train.h"

#include <bits/stdc++.h>
#define For(i, a, b) for(int i = a; i <= b; i++)
#define Forr(i, a, b) for(int i = a; i >= b; i--)
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define sz(x) ((int)x.size())
#define eb emplace_back
#define int LL
using namespace std;
using i32 = int32_t;
using LL = long long;
using pii = pair<int, int>;

const int MAXN = 5000;

vector<int> adj[MAXN + 10];
vector<int> rev[MAXN + 10];
int odeg[MAXN + 10];
int ok[MAXN + 10];

vector<i32> who_wins(vector<i32> a, vector<i32> r, vector<i32> u, vector<i32> v) {
    int n = sz(a);
    int m = sz(u);

    For(i, 0, m - 1) {
        if(u[i] == v[i] && r[v[i]] == 0) continue;
        adj[u[i]].eb(v[i]);
        rev[v[i]].eb(u[i]);
        odeg[u[i]]++;
    }
    int rt = -1;
    For(i, 0, n - 1) if(r[i]) {
        rt = i;
    }

    memset(ok, 0, sizeof(ok));
    queue<int> que;
    que.emplace(rt);
    ok[rt] = 1;
    while(!que.empty()) {
        int cur = que.front(); que.pop();
        for(auto &i:rev[cur]) if(ok[i] == 0) {
            ok[i] = 1;
            que.emplace(i);
        }
    }

    For(i, 0, n - 1) if(!ok[i]) que.emplace(i);
    while(!que.empty()) {
        int cur = que.front(); que.pop();
        for(auto &i:rev[cur]) if(ok[i]) {
            odeg[i]--;
            if(a[i] == 0 || (a[i] == 1 && odeg[i] == 0)) {
                ok[i] = 0;
                que.emplace(i);
            }
        }
    }

    if(!ok[rt]) {
        For(i, 0, n - 1) ok[i] = 0;
    }

    vector<i32> res(n);
    For(i, 0, n - 1) res[i] = (i32) ok[i];
    return res;
}

/*

2 4
0 1
1 0
0 0
0 1
1 0
1 1

1 1

*/

• Subtask 1: 0 / 5
• Subtask 2: 0 / 10
• Subtask 3: 0 / 11
• Subtask 4: 0 / 11
• Subtask 5: 0 / 12
• Subtask 6: 0 / 51