Codeforces Round #754 (Div. 2) D. Treelabeling

题目链接：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;
}