oj.uz

Submission #424130

# Submission time Handle Problem Language Result Execution time Memory
424130 2021-06-11T17:06:18 Z cfalas Toy Train (IOI17_train) C++14
0 / 100
 146 ms 50820 KB 
#include "train.h"
#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define INF 10000000
#define MOD 1000000007
#define MID ((l+r)/2)
#define HASHMOD 2305843009213693951
#define ll long long
#define ull unsigned long long
#define F first
#define S second
typedef pair<ll, ll> ii;
typedef pair<ii, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef map<int, int> mii;

#define EPS 1e-6
#define FOR(i,n) for(int i=0;i<((int)(n));i++)
#define FORi(i,a,b) for(int i=((int)(a));i<((int)(b));i++)
#define FOA(v, a) for(auto &v : a)
#define len(x) ((int)x.size())

int c=0;
vector<vi> adj;
vector<vi> rev;
vi vis;

vector<int> path;
vii loops;
vector<vi> paths;
void dfs(int s){
	path.push_back(s);
	vis[s] = c;
	paths[s] = path;
	for(auto v : adj[s]){
		if(!vis[v]) dfs(v);
		else if(vis[v]==c){
			loops.push_back({s,v});
		}
	}
	path.erase(path.end()-1);
}

vi who_wins(vi a, vi r, vi u, vi v) {
	int n = len(a);
	adj.assign(n, vi());
	rev.assign(n, vi());
	paths.resize(n);
	vis.assign(n,0);
	vi res(n, 0);

	FOR(i, len(u)){
		cout<<u[i]<<" "<<v[i]<<endl;
		adj[u[i]].push_back(v[i]);
		rev[v[i]].push_back(u[i]);
	}
	FOR(i,n){
		if(!vis[i]){
			c++, dfs(i);
		}
	}

	FOA(loop, loops){
		bool start = true;
		bool win = false;
		vi path;
		FOA(v,paths[loop.F]){
			if(v==loop.S) start = true;
			if(start){
				if(r[v]) win = true;
				path.push_back(v);
			}
		}
		if(win && !res[loop.F]){
			queue<int> q;
			FOA(v,path) q.push(v);
			vi used(n, 0);
			while(!q.empty()){
				int t = q.front();
				q.pop();
				res[t] = 1;
				FOA(v,rev[t]){
					if(!used[v]){
						used[v] = true;
						q.push(v);
					}
				}
			}
		}
	}

	return res;
}

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 136 ms 50820 KB secret mismatch
2 Halted 0 ms 0 KB -

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 204 KB secret mismatch
2 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 64 ms 23060 KB secret mismatch
2 Halted 0 ms 0 KB -

Subtask #4
0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 95 ms 22740 KB secret mismatch
2 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 146 ms 26920 KB secret mismatch
2 Halted 0 ms 0 KB -

Subtask #6
0 / 51.0

# Verdict Execution time Memory Grader output
1 Incorrect 136 ms 50820 KB secret mismatch
2 Halted 0 ms 0 KB -