CodeForces - 734E Anton and Tree

博主的 BiBi 时间

博主还有 $1\mathtt{h}+11\mathtt{m}\mathtt{i}\mathtt{n}$ 就ヾﾉ≧∀≦)o 放假啦！好鸡冻！！！

然鹅博主这个小蒟蒻在假期还有作业，但是和平时比较假期作业显得极其划算。hihihihihihihi~

Solution

感觉这题非常巧。

我们将同色的连通块缩成一个点，然后就有一个非常神奇的性质：$ans$ 就是 $\lfloor \frac{新树的直径+1}{2}\rfloor$。

你可能觉得答案应该是 $\min (黑点数,白点数)$（主要是博主先开始是这样想的），但是这样忽略了一个问题：在改变颜色的时候会增加新的连通块。

这个看似错误的性质就是在这个基础上证明的。

我们选取直径最中间的点开始（直径长度为偶就中间两个任选）。变一次色就会和相邻节点颜色相同，而相邻节点再变色会和其相邻节点相同，旁边的枝桠也可以拓展…因为选取的是直径是最长的，所以其他的部分一定会在直径统一前统一，故证。

其实真实写法并不需要用到缩点。我们设 $w(u,v)=a[u]$ $xor$ $a[v]$，$f[u]$ 定义为在其子树内以 $u$ 为起点的最长异色链的长度（就相当于缩点）。先用 $f[u]+f[v]+w(u,v)$ 更新树的直径（$f[u]$ 此时只是已经遍历过儿子的最优解，这样是正确的是因为直径在树上肯定有一个转折点或端点统计到它），再用 $f[v]+w(u,v)$ 更新 $f[u]$ 即可。

Code

#include <cstdio>
#include <iostream>
using namespace std;

const int N = 2e5 + 5;

int n, a[N], nxt[N << 1], to[N << 1], head[N], cnt, w[N << 1], f[N], ans;

int read() {
	int x = 0, f = 1; char s;
	while((s = getchar()) > '9' || s < '0') if(s == '-') f = -1;
	while(s >= '0' && s <= '9') x = (x << 1) + (x << 3) + (s ^ 48), s = getchar();
	return x * f;
}

void addEdge(const int u, const int v, const int val) {
	nxt[++ cnt] = head[u], to[cnt] = v, head[u] = cnt, w[cnt] = val;
}

void DP(const int u, const int fa) {
	for(int i = head[u]; i; i = nxt[i]) {
		int v = to[i];
		if(v == fa) continue;
		DP(v, u);
		ans = max(ans, f[u] + f[v] + w[i]);
		f[u] = max(f[u], f[v] + w[i]);
	}
}

int main() {
	n = read();
	for(int i = 1; i <= n; ++ i) a[i] = read();
	int u, v;
	for(int i = 1; i < n; ++ i) {
		u = read(), v = read();
		addEdge(u, v, a[u] ^ a[v]), addEdge(v, u, a[u] ^ a[v]);
	}
	DP(1, 0);
	printf("%d\n", ans + 1 >> 1);
	return 0;
}