题目链接:https://codeforces.com/contest/1605/problem/D
首先很容易推出,只有二进制数最高位相同的两个数,异或和会小于两数最小值,所以要让树中相邻节点的二进制数最高位不同。
考虑类似二分图,黑白两种颜色,这个题可以看作是有多种颜色,按顺序来,每个颜色的个数分别为1,2,4,…
实现是计算黑节点和白节点个数,然后计算出每种颜色的个数,因为最后一种颜色个数可能不能达到2的幂次,所以还进行了排序,然后把这些颜色填充到黑节点或白节点。按颜色个数从大到小填充,采用贪心策略,要尽量使得剩下的黑节点和白节点的最小值最大,所以一定是填充剩余数多的那个。
对一定可以实现相邻节点的值二进制数最高位1不同的思考:
假设最后一种颜色个数k满足2的幂次,设黑节点个数为x,则白节点个数为n-x,如果x<k,因为n=1+2+4+…+k,n-k=k-1,所以n-x>=k能满足该颜色的填充。填充后变为x-k和n-x,重新把倒数第二个颜色看作最后一个颜色,n’为n-k,x‘为x-k,那么会有新的x’和n’-x’,形式和开始是一样的,依然可以继续填充。
如果最后一种颜色个数k不是2的幂次,那么可以把不足的个数添加进来,树的结点个数同步增加(可随意增加结点直到个数满足最后一种颜色为2的幂次的状态),然后再按上面的方法填充,完成之后,把添加的结点再删除,可以发现还是一颗合法且满足要求的树。
显然,只要填充黑白节点剩余数大的那个,就一定可以满足填充需求,不会出现节点数不足的现象。
#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
int t, n, w[N], pos[2];
vector<int> e[N], b[2];
void dfs(int u, int f, int o) {
b[o].push_back(u);
for (int i = 0; i < e[u].size(); i++) {
int v = e[u][i];
if (v == f) continue;
dfs(v, u, o ^ 1);
}
}
struct Node {
int i, num;
bool operator < (const Node& rhs) {
return num < rhs.num;
}
};
int main(void) {
// freopen("in.txt", "r", stdin);
scanf("%d", &t);
while (t--) {
scanf("%d", &n);
for (int i = 1; i <= n; i++) e[i].clear();
for (int i = 1; i < n; i++) {
int u, v;
scanf("%d%d", &u, &v);
e[u].push_back(v);
e[v].push_back(u);
}
vector<Node> hav;
for (int i = 0; (1 << i) <= n; i++) {
int l = 1 << i, r = min(n, (1 << (i + 1)) - 1);
hav.push_back({i, r - l + 1});
}
sort(hav.begin(), hav.end());
b[0].clear();
b[1].clear();
int pos[2] = {0};
dfs(1, -1, 0);
for (int i = hav.size() - 1; i >= 0; i--) {
vector<int>& now = b[0].size() - pos[0] > b[1].size() - pos[1] ? b[0] : b[1];
int &p = b[0].size() - pos[0] > b[1].size() - pos[1] ? pos[0] : pos[1];
for (int j = 0; j < hav[i].num; j++) {
w[now[p++]] = (1 << hav[i].i) + j;
}
}
for (int i = 1; i <= n; i++)
printf("%d%c", w[i], i == n ? '\n' : ' ');
}
return 0;
}