Submission #822928

#	Time	Username	Problem	Language	Result	Execution time	Memory
822928		mindiyak	Toy Train (IOI17_train)	C++14	0 / 100	2065 ms	262144 KiB

train

#include "train.h"
using namespace std;

vector<vector<int>> paths(1e4,vector<int> ());
vector<int> chargers;
// vector<int> cost(1e4,-1);

int loop = 0;
int n;

void dfs(int pos,int start,int coins,int charging){
	if(loop==1)return;
	if(chargers[pos] == 1){
		charging = 1;
		coins = n;
	}
	if(pos==start and charging==1){loop = 1;}
	coins--;
	if(coins <= 0)return;
	for(int a:paths[pos]){
		dfs(a,start,coins,charging);
	}
}

std::vector<int> who_wins(std::vector<int> a, std::vector<int> r, std::vector<int> u, std::vector<int> v) {
	n=a.size();
	chargers = a;
	for (int i = 0; i < u.size(); i++){
		paths[u[i]].push_back(v[i]);
	}
	
	vector<int> ans(n,-1);

	
	for (int i = 0; i < u.size(); i++){
		loop = 0;
		dfs(i,i,n,chargers[i]);
		ans[i] = loop;
	}

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

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