Submission #805970

#	Time	Username	Problem	Language	Result	Execution time	Memory
805970		fatemetmhr	Toy Train (IOI17_train)	C++17	100 / 100	268 ms	1624 KiB

train

//  ~ Be Name Khoda ~  //
 
#include "train.h"
#include <bits/stdc++.h>
//#pragma GCC optimize ("O3")
//#pragma GCC target("avx2")
//#pragma GCC optimize("unroll-loops,Ofast")
 
using namespace std;
 
typedef long long ll;
 
#define pb       push_back
#define mp       make_pair
#define all(x)   x.begin(), x.end()
#define fi       first
#define se       second
 
const int maxn  =  1e6   + 10;
const int maxn5 =  5e3   + 10;
const int maxnt =  1.2e6 + 10;
const int maxn3 =  1e3   + 10;
const int mod   =  1e9   + 7;
const ll  inf   =  1e18;
 
int cnt[maxn5], q[maxn5];
vector <int> adj[maxn5], jda[maxn5], ret;
bool good[maxn5];
int n, m;

 
std::vector<int> who_wins(std::vector<int> a, std::vector<int> ch, std::vector<int> u, std::vector<int> v) {
	n = a.size(); 
	m = u.size();
 	for(int i = 0; i < m; i++){
 		adj[u[i]].pb(v[i]);
 		jda[v[i]].pb(u[i]);
 	}
 	bool con = true;
 	while(con){
 		con = false;
 		memset(good, false, sizeof good);
 		int l = 0, r = 0;
 		for(int i = 0; i < n; i++){
 			cnt[i] = adj[i].size();
 			if(ch[i])
 				q[r++] = i;
 		}
 		while(l < r){
 			int v = q[l++];
 			for(auto u : jda[v]) if(!good[u]){
 				if(a[u] || cnt[u] == 1){
 					good[u] = true;
 					if(!ch[u])
 						q[r++] = u;
 				}
 				else
 					cnt[u]--;
 			}
 		}
 		for(int i = 0; i < n; i++) if(ch[i] && !good[i]){
 			ch[i] = false;
 			con = true;
 		}
 	}
 	for(int i = 0; i < n; i++)
 		ret.pb(good[i]);
 	return ret;
}
 
 
 
 
 
 
 
 
 
 
 

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