Tree Cutting POJ - 2378 （树形DP）

题目链接：POJ - 2378

题目大意：给你n个点，然后问你这n个点中 ，去除哪些点能够使得剩下的图中最大的连通块中点的个数不超过n/2.

具体思路：第一遍dfs记录每一个点代表的子树大小，第二遍dfs记录每一个点去除的情况下，所剩余的图中最大的联通块中点的个数。

dp[i][0]代表当前i点所代表的子树大小-1

dp[i][1]代表当前i点去除的情况下，所剩余的图中最大的连通块中点的个数。

AC代码：

 #include<iostream>
 #include<stdio.h>
 #include<cmath>
 #include<algorithm>
 #include<string>
 #include<cstring>
 #include<vector>
 using namespace std;
 # define ll long long
 const int maxn = 2e5+;
 # define inf 0x3f3f3f3f
 int dp[maxn][];
 int vis[maxn];
 vector<int>Edge[maxn];
 int n;
 void init(){
 for(int i=;i<=n;i++){vis[i]=;}
 }
 void dfs1(int u){
 vis[u]=;
 for(int i=;i<Edge[u].size();i++){
 int to=Edge[u][i];
 if(vis[to])continue;
 dfs1(to);
 dp[u][]+=dp[to][];
 }
 }
 void dfs2(int u){
 vis[u]=;
 dp[u][]=n-dp[u][];
 for(int i=;i<Edge[u].size();i++){
 int to=Edge[u][i];
 if(vis[to])continue;
 dfs2(to);
 dp[u][]=max(dp[u][],dp[to][]);
 }
 }
 int main()
 {
     int st,ed;
     scanf("%d",&n);
     for(int i=;i<=n;i++){dp[i][]=;}
     for(int i=; i<n; i++)
     {
         scanf("%d %d",&st,&ed);
         Edge[st].push_back(ed);
         Edge[ed].push_back(st);
     }
     init();
     dfs1();
     init();
     dfs2();
     int flag=;
     for(int i=; i<=n; i++)
     {
         if(dp[i][]*<=n)
         {
             flag=;
             printf("%d\n",i);
         }
     }
     if(flag)
         printf("NONE\n");
     return ;
 }