poj 2378(树形dp)

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

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

 1 #include<iostream>

 2 #include<cstdio>

 3 #include<cstring>

 4 #include<algorithm>

 5 #include<vector>

 6 using namespace std;

 7 #define MAXN 11111

 8 #define FILL(a,b) memset(a,b,sizeof(a))

 9 

10 int n,num[MAXN];

11 bool mark[MAXN];

12 vector<vector<int> >g;

13 

14 int dfs(int u,int father)

15 {

16     num[u]=1;

17     bool flag=true;

18     for(int i=0;i<g[u].size();i++){

19         int v=g[u][i];

20         if(v==father)continue;

21         int tmp=dfs(v,u);

22         if(tmp>n/2)flag=false;

23         num[u]+=num[v];

24     }

25     if(flag&&n-num[u]<=n/2)mark[u]=true;

26     return num[u];

27 }

28 

29 int main()

30 {

31     int u,v;

32     while(~scanf("%d",&n)){

33         g.clear();

34         g.resize(n+2);

35         for(int i=1;i<n;i++){

36             scanf("%d%d",&u,&v);

37             g[u].push_back(v);

38             g[v].push_back(u);

39         }

40         FILL(mark,false);

41         dfs(1,-1);

42         for(int i=1;i<=n;i++)if(mark[i])printf("%d\n",i);

43     }

44     return 0;

45 }

View Code