oj.uz

Submission #798050

# Submission time Handle Problem Language Result Execution time Memory
798050 2023-07-30T10:22:40 Z LittleCube Toy Train (IOI17_train) C++17
0 / 100
 7 ms 3668 KB 
#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])
			dfs(i);
	reverse(order.begin(), order.end());
	for (int i = 0; i < n; i++)
		vis[i] = 0;
	for (int i : order)
		if (!vis[i])
			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])
			dfs(i);
	reverse(order.begin(), order.end());
	for (int i = 0; i < n; i++)
		vis[i] = 0;
	for (int i : order)
		if (!vis[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 B -> B or A
	vector<int> loop(n, 0);
	for (int i = 0; i < n; i++)
		if (a[i] == 0)
		{
			vector<int> req(n, 0);
			for (int j = 0; j < n; j++)
				if (r[j] || ans[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);
			}
			ans[i] = req[i] > 0;
		}
 
 
	// 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.0

# Verdict Execution time Memory Grader output
1 Runtime error 5 ms 2388 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 672 KB 3rd lines differ - on the 1st token, expected: '0', found: '1'
2 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Runtime error 6 ms 3668 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #4
0 / 11.0

# Verdict Execution time Memory Grader output
1 Runtime error 6 ms 2388 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Runtime error 7 ms 3156 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #6
0 / 51.0

# Verdict Execution time Memory Grader output
1 Runtime error 5 ms 2388 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -