(POJ) 2378 Tree Cutting

传送门

题目大意：给定一棵无向树，节点数为n（n<=10000），问删除那些节点可以使得新图中的每一个连通分支的节点数都不超过n/2

解题思路：这是一道入门的很裸的树型dp。dp[i]代表删除i节点后，分成的各个部分中 节点数量最大值。

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=1e4+5;
int n,dp[maxn],x,y;
vector<int> v[maxn];
bool vis[maxn];
int dfs(int p) {
	vis[p]=true;
	dp[p]=-1;
	int sum=1,tmp,mson=-1;
	for(int i=0; i<(int)v[p].size(); ++i) {
		int np=v[p][i];
		if(!vis[np]) {
			tmp=dfs(np);
			mson=max(tmp,mson);//孩子中节点最多的那个
			sum+=tmp;
			vis[np]=true;
		}
	}
	dp[p]=max(n-sum,mson);
	return sum;
}
int main() {
	scanf("%d",&n);
	for(int i=1; i<=n-1; ++i) {
		scanf("%d%d",&x,&y);
		v[x].push_back(y);
		v[y].push_back(x);
	}
	dfs(1);
	bool flag=false;
	for(int i=1; i<=n; ++i) {
		if(dp[i]*2<=n) {
			flag=true;
			printf("%d\n",i);
		}
	}
	if(!flag) {
		printf("NONE\n");
	}
	return 0;
}