Submission #390170

#	Time	Username	Problem	Language	Result	Execution time	Memory
390170		alishahali1382	장난감 기차 (IOI17_train)	C++14	100 / 100	417 ms	1444 KiB

train

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

typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
#define debug(x) {cerr<<#x<<"="<<x<<"\n";}
#define debug2(x, y) {cerr<<#x<<", "<<#y<<" = "<<x<<", "<<y<<"\n";}
#define pb push_back
#define all(x) x.begin(), x.end()

const int inf=1000000100;
const int MAXN=5010;

int n, m, k;
int Q[MAXN], L, R;
bool mark[MAXN];
int deg[MAXN];
vector<int> G[MAXN], GR[MAXN];

vector<int> who_wins(vector<int> typ, vector<int> good, vector<int> U, vector<int> V){
	n=typ.size();
	m=U.size();
	vector<int> ans(n, 1);
	for (int i=0; i<m; i++){
		G[U[i]].pb(V[i]);
		GR[V[i]].pb(U[i]);
	}
	int sz=n;
	while (sz){
		memset(mark, 0, sizeof(mark));
		L=R=0;
		// debug("check1")
		for (int i=0; i<n; i++) if (ans[i]){
			if (good[i]) Q[R++]=i, mark[i]=1;
			else{
				deg[i]=0;
				for (int j:G[i]) deg[i]+=ans[j];
			}
		}
		// debug("check2")
		while (L<R){
			int v=Q[L++];
			for (int u:GR[v]) if (ans[u] && !mark[u]){
				deg[u]--;
				if (typ[u] || !deg[u]){
					Q[R++]=u;
					mark[u]=1;
				}
			}
		}
		if (R==sz) break ;
		// debug("check3")
		L=R=0;
		for (int i=0; i<n; i++) if (ans[i]){
			if (!mark[i]) Q[R++]=i;
			else{
				deg[i]=0;
				for (int j:G[i]) deg[i]+=ans[j];
			}
		}
		while (L<R){
			int v=Q[L++];
			sz--;
			ans[v]=0;
			for (int u:GR[v]) if (ans[u] && mark[u]){
				deg[u]--;
				if (!typ[u] || !deg[u]){
					Q[R++]=u;
					mark[u]=0;
				}
			}
		}
	}

	return ans;
}

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