Submission #56624

#	Time	Username	Problem	Language	Result	Execution time	Memory
56624		Crown	Toy Train (IOI17_train)	C++14	5 / 100	26 ms	2616 KiB

train

#include "train.h"
#include <bits/stdc++.h>
using namespace std;
#define X first
#define Y second
#define pb push_back
typedef long long ll;
typedef pair<int, int> ii;

vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v)
{
	vector<int> has_cycle;
	vector<int> ending;
	int n = a.size();
	int m = u.size();
	ending.assign(n, 1);
	has_cycle.assign(n, 0);
	for(int i = 0; i< (int) m; i++)
	{
		if(u[i] != v[i])
			ending[u[i]] = false;
		else
			has_cycle[u[i]] = true;
	}
	vector<int> res(n);
	for(int i = n-1; i>= 0; i--)
	{
		if(a[i])
		{
			if(has_cycle[i])
			{
				if(r[i])
					res[i] = true;
				else
				{
					if(ending[i]) 
						res[i] = false;
					else 
						res[i] = res[i+1];
				}
			}
			else
				res[i] = res[i+1];
		}
		else
		{
			if(has_cycle[i])
			{
				if(!r[i])
					res[i] = false;
				else
				{
					if(ending[i])
						res[i] = true;
					else
						res[i] = res[i+1];
				}
			}
			else
				res[i] = res[i+1];
		}
	}
	return res;
}

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