POJ2378【树形DP】

最新推荐文章于 2020-04-17 21:31:51 发布

原创最新推荐文章于 2020-04-17 21:31:51 发布 · 479 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#poj #dp

DP 专栏收录该内容

134 篇文章

订阅专栏

本文介绍了一种在树形结构中寻找特定类型的节点的方法。通过深度优先搜索算法遍历树形结构，计算以每个节点为根的子树的节点数量，并找出删除后使剩余分支节点数量不超过总数一半的节点。

题意：

给一颗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;
}