C. Infected Tree（思维/dfs/树形dp）

题目

题意：

给定一颗树 ，根节点为2，每个结点的子节点数不超过2（也就是说这是一个二叉树）。现在根节点被感染了，我们需要拯救这棵树。
有n次过程，每次过程如下：

• 我们可以选择砍掉一个非感染点，使得该节点与周边相连的点都断开。也可以选择什么都不做。
• 被感染的点，会将与其相连的点都感染掉（这里可以忽略原本已经被感染的点）。

我们应该如何操作，才能保证最多的点（被砍掉的点不能算）。

思路：

由于每个节点都最多只有两个子节点，且感染源在根节点，我们可以通过砍掉其左节点或右节点来引导其感染的方向。
由于我们要保住更多的点，也就是让感染源最终到达的路径越短越好。
答案即为总节点数-树的最短路径-需要砍掉的最少的节点。

这里需要砍掉的节点数等于最终被感染的点数，但注意特殊情况，如果一个节点的左右子节点都为空，那么这时就不需要砍掉节点了。

#include<bits/stdc++.h> 
using namespace std;
#define ll long long
const int maxn = 300010;

int n;
vector<int> ve[maxn], deep[maxn];
// dfs返回树的最短路径 
int dfs(int u, int fa, int len) {
	deep[len].push_back(u);// deep记录深度为len的所有结点 
	if (u == 1 && ve[u].size() < 2 ||
	    u != 1 && ve[u].size() < 3) {
	    	return len;
	}
	int res = n;
	for (auto v: ve[u]) {
		if (v == fa) {
			continue;
		}
		res = min(res, dfs(v, u, len + 1));
	}
	return res;
}
void solve() {
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) {
		ve[i].clear();
		deep[i].clear();
	}
	int u, v;
	for (int i = 1; i < n; ++i) {
		scanf("%d%d", &u, &v);
		ve[u].push_back(v);
		ve[v].push_back(u);
	}
//	printf("----------------\n");// debug
	int len = dfs(1, -1, 1);
	if (len == 1) {
		printf("%d\n", max(n - 2 * len, 0));
		return;
	}
	int res = n - 2 * len;
	for (auto u: deep[len]) {
		// 如果存在子节点都为空的情况，则可以少砍一个点 
		if (ve[u].size() == 1) {
			++res;
			break;
		}
	}
	printf("%d\n", res);
}
int main() {
	int t;
	scanf("%d", &t);
	while (t--) {
		solve();
	}
}