Submission #2625766


Source Code Expand

#include <bits/stdc++.h>

using namespace std;

const int N = 2005;

int n, m;
int res;
int sum[N];
int fr[N], to[N];
int h[N], p[11][N];
int color[N * 2];
vector<int> G[N], G2[N * 2];
set< pair<int, int> > S[N];
vector< pair<int, int> > vdel[N];

int lca(int u, int v) {
	if (h[u] > h[v]) swap(u, v);
	for (int i = 10; i >= 0; --i) if (h[p[i][v]] >= h[u]) v = p[i][v];
	for (int i = 10; i >= 0; --i) if (p[i][v] != p[i][u]) v = p[i][v], u = p[i][u];
	return (u == v) ? u : p[0][u];
}

void dfs(int u) {
	for (int i = 1; i <= 10; ++i) {
		p[i][u] = p[i - 1][p[i - 1][u]];
	}
	for (auto v : G[u]) if (p[0][u] != v) {
		p[0][v] = u, h[v] = h[u] + 1, dfs(v);
	}
}

void solve(int u) {
	int pmx = 0;
	for (auto v : G[u]) if (p[0][u] != v) {
		solve(v);
		if (S[v].size() == 1) res++;
		if (S[v].size() > 1) {
			res += 2;
			if (!sum[v]) {
				auto p0 = S[v].begin();
				auto p1 = ++S[v].begin();
				G2[p0 -> second].push_back(p1 -> second);
				G2[p1 -> second].push_back(p0 -> second);
				int x = p0 -> first;
				int y = p1 -> first;
				if (h[x] > h[y]) swap(x, y);
				sum[v]++, sum[y]--;
			}
		}
		sum[u] += sum[v];
		if (!pmx || S[pmx].size() < S[v].size()) pmx = v;
	} 
	if (pmx) swap(S[u], S[pmx]);
	for (auto v : G[u]) if (p[0][u] != v) {
		for (auto i : S[v]) S[u].insert(i);
	}
	for (auto i : vdel[u]) S[u].erase(i);
}

void dfscolor(int u) {
	for (auto v : G2[u]) {
		if (color[v]) {
			assert(color[v] == 3 ^ color[u]);
			continue;
		}
		color[v] = 3 ^ color[u], dfscolor(v);
	}
}

int main() {
	ios::sync_with_stdio(false);
	cin >> n >> m;
	for (int i = 1; i < n; ++i) {
		int u, v; cin >> u >> v;
		G[u].push_back(v), G[v].push_back(u);
	}
	p[0][1] = 1, dfs(1);
	for (int i = 0; i < m; ++i) {
		cin >> fr[i] >> to[i];
		if (h[fr[i]] > h[to[i]]) swap(fr[i], to[i]);
		int tmp = lca(fr[i], to[i]);
		if (tmp == fr[i]) {
			S[to[i]].insert({tmp, i * 2});
			vdel[tmp].push_back({tmp, i * 2});
		}
		else {
			S[to[i]].insert({tmp, i * 2});
			S[fr[i]].insert({tmp, i * 2 + 1});
			G2[i * 2].push_back(i * 2 + 1);
			G2[i * 2 + 1].push_back(i * 2);
			vdel[tmp].push_back({tmp, i * 2});
			vdel[tmp].push_back({tmp, i * 2 + 1});
		}
	}
	solve(1);
	for (int i = 0; i < m * 2; ++i) {
		if (!color[i]) color[i] = 1, dfscolor(i);
	}
	cout << res << '\n';
	for (int i = 0; i < m; ++i) {
		if (color[i * 2] == 1) cout << fr[i] << ' ' << to[i] << '\n';
		else cout << to[i] << ' ' << fr[i] << '\n';
	}
}

Submission Info

Submission Time
Task E - Walking on a Tree
User aome
Language C++14 (GCC 5.4.1)
Score 1500
Code Size 2513 Byte
Status AC
Exec Time 6 ms
Memory 1536 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 1500 / 1500
Status
AC × 3
AC × 86
Set Name Test Cases
Sample sample01.txt, sample02.txt, sample03.txt
All sample01.txt, sample02.txt, sample03.txt, in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt, in49.txt, in50.txt, in51.txt, in52.txt, in53.txt, in54.txt, in55.txt, in56.txt, in57.txt, in58.txt, in59.txt, in60.txt, in61.txt, in62.txt, in63.txt, in64.txt, in65.txt, in66.txt, in67.txt, in68.txt, in69.txt, in70.txt, in71.txt, in72.txt, in73.txt, in74.txt, in75.txt, in76.txt, in77.txt, in78.txt, in79.txt, in80.txt, sample01.txt, sample02.txt, sample03.txt
Case Name Status Exec Time Memory
in01.txt AC 3 ms 896 KB
in02.txt AC 5 ms 1280 KB
in03.txt AC 5 ms 1280 KB
in04.txt AC 4 ms 1280 KB
in05.txt AC 2 ms 768 KB
in06.txt AC 2 ms 768 KB
in07.txt AC 4 ms 1152 KB
in08.txt AC 2 ms 768 KB
in09.txt AC 4 ms 1024 KB
in10.txt AC 2 ms 768 KB
in11.txt AC 4 ms 1024 KB
in12.txt AC 4 ms 1280 KB
in13.txt AC 5 ms 1408 KB
in14.txt AC 4 ms 1280 KB
in15.txt AC 3 ms 1024 KB
in16.txt AC 2 ms 640 KB
in17.txt AC 2 ms 768 KB
in18.txt AC 4 ms 1024 KB
in19.txt AC 4 ms 1152 KB
in20.txt AC 4 ms 1024 KB
in21.txt AC 4 ms 1280 KB
in22.txt AC 2 ms 640 KB
in23.txt AC 2 ms 768 KB
in24.txt AC 3 ms 896 KB
in25.txt AC 3 ms 1024 KB
in26.txt AC 3 ms 1024 KB
in27.txt AC 3 ms 896 KB
in28.txt AC 3 ms 896 KB
in29.txt AC 3 ms 1024 KB
in30.txt AC 2 ms 896 KB
in31.txt AC 3 ms 896 KB
in32.txt AC 3 ms 896 KB
in33.txt AC 3 ms 896 KB
in34.txt AC 3 ms 896 KB
in35.txt AC 3 ms 896 KB
in36.txt AC 3 ms 896 KB
in37.txt AC 3 ms 896 KB
in38.txt AC 3 ms 896 KB
in39.txt AC 3 ms 896 KB
in40.txt AC 3 ms 896 KB
in41.txt AC 3 ms 768 KB
in42.txt AC 3 ms 768 KB
in43.txt AC 3 ms 896 KB
in44.txt AC 3 ms 768 KB
in45.txt AC 3 ms 896 KB
in46.txt AC 3 ms 896 KB
in47.txt AC 3 ms 896 KB
in48.txt AC 3 ms 896 KB
in49.txt AC 3 ms 896 KB
in50.txt AC 3 ms 896 KB
in51.txt AC 3 ms 896 KB
in52.txt AC 3 ms 896 KB
in53.txt AC 3 ms 896 KB
in54.txt AC 3 ms 896 KB
in55.txt AC 3 ms 768 KB
in56.txt AC 3 ms 896 KB
in57.txt AC 3 ms 896 KB
in58.txt AC 3 ms 768 KB
in59.txt AC 3 ms 896 KB
in60.txt AC 3 ms 768 KB
in61.txt AC 3 ms 1024 KB
in62.txt AC 3 ms 896 KB
in63.txt AC 3 ms 1024 KB
in64.txt AC 3 ms 1024 KB
in65.txt AC 3 ms 1024 KB
in66.txt AC 3 ms 896 KB
in67.txt AC 3 ms 1024 KB
in68.txt AC 3 ms 1024 KB
in69.txt AC 3 ms 1024 KB
in70.txt AC 3 ms 1024 KB
in71.txt AC 2 ms 768 KB
in72.txt AC 2 ms 768 KB
in73.txt AC 6 ms 1408 KB
in74.txt AC 6 ms 1536 KB
in75.txt AC 6 ms 1408 KB
in76.txt AC 6 ms 1408 KB
in77.txt AC 6 ms 1536 KB
in78.txt AC 6 ms 1536 KB
in79.txt AC 6 ms 1536 KB
in80.txt AC 6 ms 1536 KB
sample01.txt AC 1 ms 640 KB
sample02.txt AC 1 ms 640 KB
sample03.txt AC 1 ms 640 KB