给一棵树,求删除某个节点,使得剩下的最大的连通分量的最大值小于总结点的一半
树形dp
d[i][0]表示以i为根节点的树的总节点数(除了i)
d[i][1]表示以i为根节点的树的最大的子树的节点数
则若满足d[i][1] <= n/2 并且 n-1-d[i][0] <= n/2 ,节点i就是满足条件的一个
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
const int N = 10010;
int n,u,v,d[N][2];
vector<int> G[N];
void dfs(int u,int fa){
d[u][0] = 0;
d[u][1] = 0;
for(int i = 0;i<G[u].size();i++){
int v = G[u][i];
if(v == fa) continue;
dfs(v,u);
d[u][0] += d[v][0] + 1;
d[u][1] = max(d[u][1],d[v][0] + 1);
}
}
int main(){
while(~scanf("%d",&n)){
for(int i = 1;i<=n;i++) G[i].clear();
for(int i = 0; i<n-1; i++) {
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,-1);
int flag = 0;
for(int i = 1;i<=n;i++){
if(d[i][1] <= n /2 && (n - d[i][0] -1 ) <= n/2 ) {
printf("%d\n",i);
flag = 1;
}
}
if(!flag) printf("NONE\n");
}
return 0;
}