Submission #775850

#	Time	Username	Problem	Language	Result	Execution time	Memory
775850		Amylopectin	Toy Train (IOI17_train)	C++14	5 / 100	438 ms	1828 KiB

train

#include "train.h"
#include <vector>
#include <algorithm>
#include <stdio.h>

using namespace std;

// const int mxn = 20;
const int mxn = 1e4 + 10;
vector<int> pat[mxn] = {},ans;
int batt[mxn] = {},who[mxn] = {},n,u[mxn] = {},res[mxn] = {},nba[mxn] = {};
int re(int cn,int cnba)
{
	int i,j,fn,of = 0;
	u[cn] = 1;
	nba[cn] = cnba;
	cnba += batt[cn];
	if(who[cn] == 1)
	{
		for(i=0; i<pat[cn].size(); i++)
		{
			fn = pat[cn][i];
			if(u[fn] == 2)
			{
				if(res[fn] == 1)
				{
					of = 1;
					break;
				}
			}
			else if(u[fn] == 1)
			{
				if(nba[fn] < cnba)
				{
					of = 1;
					break;
				}
			}
			else 
			{
				re(fn,cnba);
				if(res[fn] == 1)
				{
					of = 1;
					break;
				}
			}
		}
		if(of == 0)
		{
			res[cn] = 0;
		}
		else 
		{
			res[cn] = 1;
		}
	}
	else 
	{
		for(i=0; i<pat[cn].size(); i++)
		{
			fn = pat[cn][i];
			if(u[fn] == 2)
			{
				if(res[fn] == 0)
				{
					of = 1;
					break;
				}
			}
			else if(u[fn] == 1)
			{
				if(nba[fn] == cnba)
				{
					of = 1;
					break;
				}
			}
			else 
			{
				re(fn,cnba);
				if(res[fn] == 0)
				{
					of = 1;
					break;
				}
			}
		}
		if(of == 0)
		{
			res[cn] = 1;
		}
		else 
		{
			res[cn] = 0;
		}
	}
	u[fn] = 2;
	return 0;
}
std::vector<int> who_wins(std::vector<int> aa, std::vector<int> rr
, std::vector<int> uu, std::vector<int> vv) 
{
	int i,j,m = uu.size(),cn,cm,fn,fm;
	n = aa.size();
	for(i=0; i<n; i++)
	{
		who[i] = aa[i];
		batt[i] = rr[i];
	}
	for(i=0 ;i<m; i++)
	{
		pat[uu[i]].push_back(vv[i]);
	}
	for(i=0; i<n; i++)
	{
		for(j=0; j<n; j++)
		{
			u[j] = 0;
		}
		re(i,0);
		ans.push_back(res[i]);
	}
	return ans;
	// std::vector<int> res(a.size());
	// for(int i = 0; i < (int)res.size(); i++)
	// 	res[i] = i % 2;
	// return res;
}

Compilation message (stderr)

train.cpp: In function 'int re(int, int)':
train.cpp:20:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |   for(i=0; i<pat[cn].size(); i++)
      |            ~^~~~~~~~~~~~~~~
train.cpp:60:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |   for(i=0; i<pat[cn].size(); i++)
      |            ~^~~~~~~~~~~~~~~
train.cpp:14:8: warning: unused variable 'j' [-Wunused-variable]
   14 |  int i,j,fn,of = 0;
      |        ^
train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:104:24: warning: unused variable 'cn' [-Wunused-variable]
  104 |  int i,j,m = uu.size(),cn,cm,fn,fm;
      |                        ^~
train.cpp:104:27: warning: unused variable 'cm' [-Wunused-variable]
  104 |  int i,j,m = uu.size(),cn,cm,fn,fm;
      |                           ^~
train.cpp:104:30: warning: unused variable 'fn' [-Wunused-variable]
  104 |  int i,j,m = uu.size(),cn,cm,fn,fm;
      |                              ^~
train.cpp:104:33: warning: unused variable 'fm' [-Wunused-variable]
  104 |  int i,j,m = uu.size(),cn,cm,fn,fm;
      |                                 ^~
train.cpp: In function 'int re(int, int)':
train.cpp:98:8: warning: 'fn' may be used uninitialized in this function [-Wmaybe-uninitialized]
   98 |  u[fn] = 2;
      |  ~~~~~~^~~

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