oj.uz

Submission #103955

# Submission time Handle Problem Language Result Execution time Memory
103955 2019-04-03T12:57:12 Z luciocf Mousetrap (CEOI17_mousetrap) C++14
0 / 100
 6 ms 1536 KB 
#include <bits/stdc++.h>

using namespace std;

const int maxn = 11;

int n, t, m;

int pai[maxn], nivel[maxn];

int dp[1<<maxn][maxn][2];

vector<pair<int, int>> grafo[maxn];

void dfs(int u, int p)
{
	for (auto pp: grafo[u])
	{
		int v = pp.first;
		if (v == p) continue;

		pai[v] = u, nivel[v] = nivel[u]+1;
		dfs(v, u);
	}
}

bool stuck(int mask, int u)
{
	for (auto v: grafo[u])
		if (v.first != pai[u] && !(mask&(1<<v.second))) return false;
	return true;
}

int lca(int u, int v)
{
	while (u != v)
	{
		if (nivel[u] > nivel[v]) u = pai[u];
		else v = pai[v];
	}

	return u;
}

int solve(int mask, int u, bool q)
{
	if (u == t) return 0;
	if (dp[mask][u][q] != -1) return dp[mask][u][q];

	if (stuck(mask, u))
	{
		int ans = 0;
		while (true)
		{
			for (auto v: grafo[u])
				if (v.first != pai[u] && !(mask&(1<<v.second)))
					ans++;	

			if (u == lca(u, t)) break;

			ans++;
			u = pai[u];
		}

		if (lca(u, t) == t) return dp[mask][u][q] = ans;

		u = t;

		while (true)
		{
			for (auto v: grafo[u])
				if (v.first != pai[u] && !(mask&(1<<v.second)))
					ans++;

			if (u == lca(u, t)) break;

			ans++;
			u = pai[u];
		}

		return dp[mask][u][q] = ans;
	}

	int ans = 0;
	if (q) ans = solve(mask, u, 0);

	for (auto v: grafo[u])
	{
		if (v.first == pai[u]) continue;
		if ((mask&(1<<v.second))) continue;

		if (q) ans = min(ans, 1+solve(mask|(1<<v.second), u, 0));
		else ans = max(ans, solve(mask|(1<<v.second), v.first, 1));
	}

	return dp[mask][u][q] = ans;
}

int main(void)
{
	scanf("%d %d %d", &n, &t, &m);

	for (int i = 0; i < n-1; i++)
	{
		int u, v;
		scanf("%d %d", &u, &v);

		grafo[u].push_back({v, i});
		grafo[v].push_back({u, i});
	}

	dfs(m, 0);

	memset(dp, -1, sizeof dp);
	printf("%d\n", solve(0, m, 1));
}

Compilation message

mousetrap.cpp: In function 'int main()':
mousetrap.cpp:101:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d %d %d", &n, &t, &m);
  ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
mousetrap.cpp:106:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d %d", &u, &v);
   ~~~~~^~~~~~~~~~~~~~~~~

Subtask #1
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 512 KB Output is correct
2 Incorrect 2 ms 512 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 25.0

# Verdict Execution time Memory Grader output
1 Runtime error 6 ms 1536 KB Execution killed with signal 11 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -

Subtask #3
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 512 KB Output is correct
2 Incorrect 2 ms 512 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 512 KB Output is correct
2 Incorrect 2 ms 512 KB Output isn't correct
3 Halted 0 ms 0 KB -