POJ2378【树形DP】

来源:互联网 发布:yum install gcc 失败 编辑:程序博客网 时间:2024/06/08 06:31

题意:

给一颗n个结点的树,节点编号为1~n,把删除一个节点之后,
剩下的分支中节点数量最多的数量小于等于总数量一半的编号全部按顺序输出。

思路:

对于节点 node ,它的值无非是前驱所有,后驱所有。
所以求以每个节点为根的子树的节点个数就好了,判断一下就好了。

Code:

//#include <bits/stdc++.h>#include<iostream>#include<vector>#include<cstdio>#include<string.h>#include<algorithm>using namespace std;typedef pair<int,int> PII;typedef long long LL;//#pragma comment(linker, "/STACK:102400000,102400000")const int N=1e4+10;struct Edge{    int v;    int next;}edge[N<<1];int head[N],tol;int n;void init(){    tol=0;    memset(head,-1,sizeof(head));}void add(int u,int v){   edge[tol].v=v;   edge[tol].next=head[u];   head[u] = tol++;}void input(){    int u,v;    scanf("%d",&n);    init();    for(int i=1;i<n;i++){        scanf("%d%d",&u,&v);        add(u,v);        add(v,u);    }}bool vis[N];int dp[N];vector<int>ans;void DFS(int u){    int v;    dp[u] = 1;    int sum = 0;    int Max = 0;    for(int i=head[u];~i;i=edge[i].next){        v = edge[i].v;        if(vis[v]) continue;        vis[v] = true;        DFS(v);        sum += dp[v];        Max = max(Max,dp[v]);    }    dp[u] = sum + dp[u];    Max = max(Max,n-dp[u]);    if(Max <= (n/2)) ans.push_back(u);}void solve(){    memset(vis,false,sizeof(vis));    vis[1]=true;    ans.clear();    DFS(1);}void print(){    sort(ans.begin(),ans.end());    int sz = ans.size();    for(int i=0;i<sz;i++)        printf("%d\n",ans[i]);}int main(){    input();    solve();    print();    return 0;}