poj 2378(树形dp)

题目链接：http://poj.org/problem?id=2378

思路：num[u]表示以u为根的子树的顶点个数（包括），如果去掉u之后u的每棵子树都小于等于n/2，则选择u。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 using namespace std;
 #define MAXN 11111
 #define FILL(a,b) memset(a,b,sizeof(a))

 int n,num[MAXN];
 bool mark[MAXN];
 vector<vector<int> >g;

 int dfs(int u,int father)
 {
     num[u]=;
     bool flag=true;
     for(int i=;i<g[u].size();i++){
         int v=g[u][i];
         if(v==father)continue;
         int tmp=dfs(v,u);
         if(tmp>n/)flag=false;
         num[u]+=num[v];
     }
     if(flag&&n-num[u]<=n/)mark[u]=true;
     return num[u];
 }

 int main()
 {
     int u,v;
     while(~scanf("%d",&n)){
         g.clear();
         g.resize(n+);
         for(int i=;i<n;i++){
             scanf("%d%d",&u,&v);
             g[u].push_back(v);
             g[v].push_back(u);
         }
         FILL(mark,false);
         dfs(,-);
         for(int i=;i<=n;i++)if(mark[i])printf("%d\n",i);
     }
     return ;
 }