#include "train.h"
#include<bits/stdc++.h>
using namespace std;
vector<int> who_wins(vector<int> Own, vector<int> Charge, vector<int> U, vector<int> V) {
int N = Own.size();
vector<vector<int>> AR(N), AR2(N);
int M = U.size();
for (int i=0; i<M; i++)
{
AR[U[i]].push_back(V[i]);
AR2[V[i]].push_back(U[i]);
}
vector<int> occ(N);
while(true)
{
vector<int> left(N);
occ.assign(N, 0);
for (int i=0; i<N; i++)left[i]=AR[i].size();
queue<int> Q;
for (int i=0; i<N; i++)
{
if (Charge[i]==0)continue;
Q.push(i);
occ[i]=1;
}
while(!Q.empty())
{
int x = Q.front();
Q.pop();
for (int i=0; i<AR2[x].size(); i++)
{
if (occ[AR2[x][i]])continue;
left[AR2[x][i]]--;
if (Own[AR2[x][i]]==1 || left[AR2[x][i]]==0)
{
occ[AR2[x][i]]=1;
Q.push(AR2[x][i]);
}
}
}
int rem=0;
for (int i=0; i<N; i++)
{
if (Charge[i]==0)continue;
int cnt=0;
for (int j=0; j<AR[i].size(); j++)
cnt+=occ[AR[i][j]];
if (cnt==0 || (cnt<AR[i].size() && Own[i]==0)){
rem=1;
Charge[i]=0;
}
}
if (rem==0)break;
}
return occ;
}
