AT5759 ThREE（构造）

Description

有一棵 $n$ 个点的无根树，边权为一。求一个 $1\sim n$ 的排列 $p_1\sim p_n$，使得对于任意两个下标 $(i,j)$ 如果点 $i$ 与点 $j$ 在树上的距离为 $3$ 那么 $p_i+p_j|3\vee p_i{p}_j|3$。有多个随意输出一个，如果没有输出 $-1$。

$1\le n\le 2\times 10^5$。

Solution

根据样例，不可能没有方案。

令 $1$ 为根节点，将 $1\sim n$ 的每个数 $\mathrm{m}\mathrm{o}\mathrm{d}3$。因为一个数加 $\pm 3$ 它的奇偶性一定改变，不妨用黑白染色的方法，将每个点按与根节点距离的奇偶性分类。

对于一个数对 $p_i,p_j$ 假设 $p_i\le p_j$。如果 $i$ 和 $j$ 的距离为 $3$ 那么有 p i = 1 , p j = 2 ∨ p i = 0 , p j = { 0 , 1 , 2 } p_i = 1,p_j = 2 \lor p_i = 0, p_j = \{ 0,1,2 \} pi​=1,pj​=2∨pi​=0,pj​={0,1,2}。

所以构造方案非常简单，先在能将 $1,2$ 分完的前提下，尽量把它们分在不同奇偶性的点集中，再把 $0$ 分给余下的点。

Code

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
const int N = 2e5 + 5, INF = 0x3f3f3f3f;
inline int read() {
	int x = 0, f = 0; char ch = 0;
	while (!isdigit(ch)) f |= ch == '-', ch = getchar();
	while (isdigit(ch)) x = (x << 3) + (x << 1) + (ch ^ 48), ch = getchar();
	return f ? -x : x;
}
int ans[N];
vector <int> G[N], col[2], mod[3];
void dfs(int x, int c, int fa) {
	col[c].push_back(x);
	for (auto y : G[x]) if (y != fa) dfs(y, c ^ 1, x);
}
void sol(int i, int j) {
	while (!col[i].empty() && !mod[j].empty()) {
		ans[col[i].back()] = mod[j].back();
		col[i].pop_back(), mod[j].pop_back();
	}
}
int main() {
	int n = read();
	for (int i = 1; i < n; i++) {
		int x = read(), y = read();
		G[x].pb(y), G[y].pb(x);
	}
	for (int i = 1; i <= n; i++) mod[i % 3].pb(i);
	dfs(1, 0, 0);
	sol(col[0].size() < col[1].size(), 1), sol(col[0].size() < col[1].size(), 2);
	sol(0, 0), sol(1, 0);
	for (int i = 1; i <= n; i++) printf("%d ", ans[i]);
	return 0;
}