# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
607430 | 2022-07-26T17:17:25 Z | yanndev | Toy Train (IOI17_train) | C++17 | 0 ms | 0 KB |
#include <bits/stdc++.h> using namespace std; const int MX = 5042; int n; int mxScore[MX]; int deg[MX]; bool isAlice[MX]; bool isCharge[MX]; vector<int> graph[MX]; void BFS(int layer) { queue<int> nxt {}; for (int i = 0; i < n; i++) { deg[i] = 0; if (isCharge[i] && mxScore[i] == layer) nxt.push(layer); } for (int i = 0; i < n; i++) if (mxScore[i] == layer) for (auto& x: graph[i]) deg[x]++; while (!nxt.empty()) { auto cur = nxt.front(); auto chg = min; if (isAlice[cur]) chg = max; nxt.pop(); for (auto& x: graph[cur]) if (--deg[x] == 0) mxScore[cur] = chg(mxScore[cur], mxScore[x]); mxScore[cur] += isCharge[cur]; } } vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v) { n = (int)a.size(); for (int i = 0; i < (int)u.size(); i++) graph[v[i]].push_back(u[i]); for (int i = 0; i < n; i++) { isAlice[i] = a[i]; isCharge[i] = r[i]; } for (int layer = 0; layer <= (int)r.size(); layer++) BFS(layer); vector<int> ans (n); for (int i = 0; i < n; i++) ans[i] = mxScore[i] >= (int)r.size() + 1; return ans; }