Submission #775970

#	Time	Username	Problem	Language	Result	Execution time	Memory
775970		PoonYaPat	Toy Train (IOI17_train)	C++14	11 / 100	1188 ms	1524 KiB

train

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

int n;
bool c[5005],charge[5005],vis[5005];
vector<int> adj[5005],radj[5005],ans;

void dfs(int x) {
	for (auto s : adj[x]) {
		if (!vis[s]) {
			vis[s]=true;
			dfs(s);
		}
	}
}

void rdfs(int x) {
	vis[x]=true;
	ans[x]=1;
	for (auto s : radj[x]) {
		if (!vis[s]) {
			rdfs(s);
		}
	}
}

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {

	n=a.size();
	for (int i=0; i<n; ++i) charge[i]=r[i], c[i]=a[i], ans.push_back(0);

	for (int i=0; i<u.size(); ++i) {
		adj[u[i]].push_back(v[i]);
		radj[v[i]].push_back(u[i]);
	}

	for (int i=0; i<n; ++i) {
		if (!charge[i]) continue;
		memset(vis,0,sizeof(vis));
		dfs(i);
		if (!vis[i]) continue;

		memset(vis,0,sizeof(vis));
		rdfs(i);
	}

	return ans;
}

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:33:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |  for (int i=0; i<u.size(); ++i) {
      |                ~^~~~~~~~~

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