제출 #797946

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
797946		LittleCube	장난감 기차 (IOI17_train)	C++17	0 / 100	419 ms	1624 KiB

train

#include "train.h"
#include <bits/stdc++.h>
using namespace std;

vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v)
{
	int n = a.size(), m = u.size();
	vector<vector<int>> E(n), R(n);
	for (int i = 0; i < m; i++)
		E[u[i]].emplace_back(v[i]),
			R[v[i]].emplace_back(u[i]);
	vector<int> loop(n, 0);
	for (int i = 0; i < n; i++)
	{
		vector<int> req(n, 0);
		for (int j = 0; j < n; j++)
			if (r[j])
				req[j] = 1e9;
			else if (a[j])
				req[j] = E[j].size();
			else
				req[j] = 1;
		queue<int> q;
		q.push(i);
		while (!q.empty())
		{
			int x = q.front();
			q.pop();
			for (auto y : R[x])
				if (--req[y] == 0)
					q.emplace(y);
		}
		loop[i] = req[i] <= 0;
	}
	vector<int> ans(n, 1);
	vector<int> req(n, 0);
	queue<int> q;

	for (int j = 0; j < n; j++)
		if (loop[j])
		{
			q.push(j);
			req[j] = 0;
		}
		else if (r[j])
			req[j] = 1e9;
		else if (a[j])
			req[j] = E[j].size();
		else
			req[j] = 1;

	while (!q.empty())
	{
		int x = q.front();
		ans[x] = 0;
		q.pop();
		for (auto y : R[x])
			if (--req[y] == 0)
				q.emplace(y);
	}
	return ans;
}

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