Submission #607530

#	Time	Username	Problem	Language	Result	Execution time	Memory
607530		Arturgo	장난감 기차 (IOI17_train)	C++14	0 / 100	922 ms	1240 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) {
	if(nouvGagnants[sommet]) return;
	nouvGagnants[sommet] = true;
	
	for(int parent : parents[sommet]) {
		if(estMax[parent]) {
			explore(parent);
		}
		else {
			nouvDegs[parent]--;
			if(nouvDegs[parent] == 0)
				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);
	
	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 + 1;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: 0 / 11
• Subtask 4: 0 / 11
• Subtask 5: 0 / 12
• Subtask 6: 0 / 51