Submission #425279

#	Time	Username	Problem	Language	Result	Execution time	Memory
425279		flappybird	Toy Train (IOI17_train)	C++14	23 / 100	248 ms	50228 KiB

train

#include "train.h"

#include <bits/stdc++.h>
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx,avx2,fma")

using namespace std;
typedef int ll;

#define MAX 6000

vector<ll> A, s;
ll N, M;
vector<ll> adj[MAX], rev[MAX], deg, d;
ll mp[MAX][MAX];

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
	ll i;
	A = a;
	N = a.size();
	M = u.size();
	d.resize(N);
	deg.resize(N);
	vector<ll> res;
	res.resize(N);
	for (i = 0; i < N; i++) if (r[i]) s.push_back(i);
	for (i = 0; i < M; i++) adj[u[i]].push_back(v[i]), rev[v[i]].push_back(u[i]), d[u[i]]++;
	for (auto st : s) {
		deg = d;
		queue<ll> q;
		vector<ll> vis;
		vis = res;
		q.push(st);
		vis[st] = 1;
		while (!q.empty()) {
			ll t = q.front();
			q.pop();
			for (auto x : rev[t]) {
				if (x == -1) continue;
				if (mp[t][x]) continue;
				if (vis[x]) continue;
				deg[x]--;
				if (a[x]) q.push(x), vis[x] = 1;
				else if (deg[x] == 0) q.push(x), vis[x] = 1;
			}
		}
		bool c = true;
		if (!a[st]) {
			for (auto asdf : adj[st]) {
				if (!vis[asdf]) {
					c = 0;
					break;
				}
			}
		}
		else {
			c = 0;
			for (auto asdf : adj[st]) {
				if (vis[asdf]) {
					c = 1;
					break;
				}
			}
		}
		if (c) {
			for (i = 0; i < N; i++) res[i] |= vis[i];
			for (i = 0; i < adj[st].size(); i++) {
				if (vis[adj[st][i]]) mp[st][adj[st][i]] = mp[adj[st][i]][st] = 1;
			}
		}
		else {
			for (i = 0; i < adj[st].size(); i++) if (vis[adj[st][i]]) adj[st][i] = 0;
		}
	}
	return res;
}

Compilation message (stderr)

train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:69:18: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   69 |    for (i = 0; i < adj[st].size(); i++) {
      |                ~~^~~~~~~~~~~~~~~~
train.cpp:74:18: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |    for (i = 0; i < adj[st].size(); i++) if (vis[adj[st][i]]) adj[st][i] = 0;
      |                ~~^~~~~~~~~~~~~~~~

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