| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 222129 | emil_physmath | 장난감 기차 (IOI17_train) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "train.h"
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
#include <set>
using namespace std;
const int maxN = 5000;
vector<int> nei[maxN];
int charge[maxN], a[maxN];
int ans[maxN];
int n;
vector<int> rnei[maxN];
void OK(vector<char>& mask)
{
queue<int> cycle;
for (int i = 0; i < n; ++i)
if (mask[i])
cycle.push(i);
int cnt = cycle.size();
while (cycle.size())
{
int u = cycle.front();
cycle.pop();
if (!mask[u]) continue;
bool g = false;
if (a[u])
{
g = true;
for (int to: nei[u])
if (!mask[to] && ans[to] != 0)
g = false;
}
else
{
g = false;
for (int to: nei[u])
if (mask[to] || ans[to] == 0)
g = true;
}
if (!g)
{
--cnt;
mask[u] = 0;
for (int to: rnei[u])
if (mask[to])
cycle.push(to);
}
}
return cnt;
}
vector<int> Solve()
{
for (int i = 0; i < n; ++i)
ans[i] = 1;
vector<char> stmask(n);
for (int i = 0; i < n; ++i)
if (!charge[i])
stmask[i] = 1;
while (true)
{
auto mask = stmask;
if (!OK(mask))
break;
for (int i = 0; i < n; ++i)
if (mask[i])
{
ans[i] = 0;
stmask[i] = 0;
}
for (int i = 0; i < n; ++i)
for (int u = 0; u < n; ++u)
{
bool g = false;
if (a[u])
{
g = true;
for (int to: nei[u])
if (ans[to] != 0)
g = false;
}
else
{
g = false;
for (int to: nei[u])
if (ans[to] == 0)
g = true;
}
if (g)
ans[u] = 0, stmask[u] = 0;
}
}
return vector<int>(ans, ans + n);
}
std::vector<int> who_wins(std::vector<int> a_, std::vector<int> charge_, std::vector<int> u, std::vector<int> v) {
n = a_.size();
for (int i = 0; i < n; ++i)
a[i] = a_[i], charge[i] = charge_[i];
for (int i = 0; i < u.size(); ++i)
nei[u[i]].push_back(v[i]);
for (int v = 0; v < n; ++v)
for (int to: nei[v])
rnei[to].push_back(v);
return Solve();
}