Submission #607539

#	Time	Username	Problem	Language	Result	Execution time	Memory
607539		Arturgo	Toy Train (IOI17_train)	C++14	23 / 100	1341 ms	1364 KiB

train

#include "train.h"
#include <deque>
#include <vector>
using namespace std;

int nbSommets;
vector<int> nouvDegs;
vector<bool> nouvGagnants;
vector<int> estMax;

vector<int> parents[6000];

void explore(int sommet) {
	for(int parent : parents[sommet]) {
		if(nouvGagnants[parent]) continue;
		
		if(estMax[parent]) {
			nouvGagnants[parent] = true;
			explore(parent);
		}
		else {
			nouvDegs[parent]--;
			if(nouvDegs[parent] == 0) {
				nouvGagnants[parent] = true;
				explore(parent);
			}
		}
	}	
}

vector<int> who_wins(vector<int> _estMax, vector<int> estRecharge,
 vector<int> debs, vector<int> fins) {
 	estMax = _estMax;
	nbSommets = estMax.size();
	
	vector<int> degs(nbSommets, 0);
	
	for(int iArc = 0;iArc < (int)debs.size();iArc++) {
		parents[fins[iArc]].push_back(debs[iArc]);
		degs[debs[iArc]]++;
	}
	
	int nbRecharges = 0;
	for(int iSommet = 0;iSommet < nbSommets;iSommet++) {
		if(estRecharge[iSommet])
			nbRecharges += 1;
	}
	
	vector<bool> gagnants(nbSommets, true);
	
	for(int iFois = 0;iFois <= nbRecharges + 3;iFois++) {
		nouvDegs = degs;
		nouvGagnants = vector<bool>(nbSommets, false);
		
		for(int iSommet = 0;iSommet < nbSommets;iSommet++) {
			if(gagnants[iSommet] && estRecharge[iSommet]) {
				explore(iSommet);
			}
		}
		
		gagnants = nouvGagnants;
	}
	
	vector<int> estGagnant;
	for(int iSommet = 0;iSommet < nbSommets;iSommet++) {
		estGagnant.push_back(gagnants[iSommet]);
	}
	
	return estGagnant;
}

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