제출 #798054

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
798054		LittleCube	Toy Train (IOI17_train)	C++17	11 / 100	7 ms	1876 KiB

train

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

int n, m, k, vis[5005], scc[5005], self[5005];
vector<int> E[5005], R[5005];
vector<int> order;
vector<int> cc[5005];

void dfs(int u)
{
	vis[u] = 1;
	for (auto v : E[u])
		if (!vis[v])
			dfs(v);
	order.emplace_back(u);
}

void dfscc(int u, int c)
{
	vis[u] = 1;
	scc[u] = c;
	cc[c].emplace_back(u);
	for (auto v : R[u])
		if (!vis[v])
			dfscc(v, c);
}

vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v)
{
	n = a.size(), m = u.size(), k = -1;
	for (int i = 0; i < n + 1; i++)
		vis[i] = 0, E[i].clear(), R[i].clear(), cc[i].clear(), self[i] = 0;

	// Check solely A
	order.clear();
	for (int i = 0; i < m; i++)
	{
		if (a[u[i]] == 1 && a[v[i]] == 1)
		{
			E[u[i]].emplace_back(v[i]);
			R[v[i]].emplace_back(u[i]);
		}
		if (u[i] == v[i])
			self[u[i]] = 1;
	}

	for (int i = 0; i < n; i++)
		if (!vis[i] && a[i] == 1)
			dfs(i);
	reverse(order.begin(), order.end());
	for (int i = 0; i < n; i++)
		vis[i] = 0;
	for (int i : order)
		if (!vis[i] && a[i] == 1)
			dfscc(i, ++k);

	vector<int> ans(n, 0);

	for (int i = k; i >= 0; i--)
	{
		int flag = 0;
		for (int j : cc[i])
		{
			for (int p : E[j])
				if (ans[p])
					flag = 1;
			if (r[j] == 1 && (cc[i].size() > 1 || self[j]))
				flag = 1;
		}
		if (flag)
			for (int j : cc[i])
				ans[j] = 1;
	}

	// Check B (no charging stations)
	for (int i = 0; i < n + 1; i++)
		vis[i] = 0, E[i].clear(), R[i].clear(), cc[i].clear(), self[i] = 0;
	order.clear();
	for (int i = 0; i < m; i++)
	{
		if (a[u[i]] == 0 && a[v[i]] == 0 && !r[u[i]] && !r[v[i]])
		{
			E[u[i]].emplace_back(v[i]);
			R[v[i]].emplace_back(u[i]);
			if (u[i] == v[i])
				self[u[i]] = 1;
		}
	}

	for (int i = 0; i < n; i++)
		if (!vis[i] && a[i] == 0 && !r[i])
			dfs(i);
	reverse(order.begin(), order.end());
	for (int i = 0; i < n; i++)
		vis[i] = 0;
	for (int i : order)
		if (!vis[i] && a[i] == 0 && !r[i])
			dfscc(i, ++k);

	for (int i = k; i >= 0; i--)
	{
		int flag = 0;
		for (int j : cc[i])
		{
			for (int p : E[j])
				if (ans[p])
					flag = 1;
			if (r[j] == 1 && (cc[i].size() > 1 || self[j]))
				flag = 1;
		}
		if (!flag)
			for (int j : cc[i])
				ans[j] = 1;
	}

	// check mixed
	for (int i = 0; i < n + 1; i++)
		E[i].clear(), R[i].clear();
	for (int i = 0; i < m; i++)
	{
		E[u[i]].emplace_back(v[i]);
		R[v[i]].emplace_back(u[i]);
	}

	vector<int> req(n, 0);
	queue<int> q;

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

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

	return ans;
}

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