Submission #426462

#	Time	Username	Problem	Language	Result	Execution time	Memory
426462		jeqcho	Toy Train (IOI17_train)	C++17	0 / 100	11 ms	1484 KiB

train

#include "train.h"
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<int,int> pii;
typedef vector<int> vi;
typedef vector<pii> vpi;

#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define F0R(i,b) FOR(i,0,b)
#define ROF(i,a,b) for(int i=(b)-1;i>=(a);--i)
#define R0F(i,b) ROF(i,0,b)
#define all(x) begin(x),end(x)
#define sz(x) int(x.size())
#define pb push_back
#define rsz resize
#define trav(a,x) for(auto&a :x)
#define fi first
#define se second

int const N=5e3+3;
vi to[N];
vi tor[N];
vi r;
int n;
bitset<N> vis;
stack<int> s;
vi topsort;

void dfs(int now)
{
	if(vis[now])return;
	vis[now]=1;
	trav(chi,to[now])
	{
		dfs(chi);
	}
	s.push(now);
}

void dfs2(int now)
{
	if(vis[now])return;
	vis[now]=1;
	trav(chi,tor[now])
	{
		dfs2(chi);
	}
	topsort.pb(now);
}

void kos()
{
	F0R(i,n)
	{
		dfs(i);
	}
	vis.reset();
	while(!s.empty())
	{
		int u=s.top();
		s.pop();
		dfs2(u);
	}
	reverse(all(topsort));
}

vi who_wins(vi a, vi r1, vi u, vi v) {
	n=sz(a);
	r=r1;
	F0R(i,n)
	{
		to[u[i]].pb(v[i]);
		tor[v[i]].pb(u[i]);
	}
	kos();
	bool ok=0;
	vi ans(n);
	R0F(i,n)
	{
		if(r[topsort[i]])
		{
			ok=1;
		}
		ans[topsort[i]]=ok;
	}
	return ans;
}

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