poj 2378 Tree Cutting(树形DP,删点使得独立的部分结点数不超过n/2)

来源:互联网 发布:网络教育的意义 编辑:程序博客网 时间:2024/05/21 16:23

1、http://poj.org/problem?id=2378

2、题目大意:

一棵树有n个结点,现在要删除某个结点,使得分成的部分结点数最多不超过n/2个,题目与poj 3107类似,只是求得问题不同

用一个cnt[]数组记录下每个点有多少个子节点

那么我们要求的删除根节点u后剩余部分的最大数就是要么是u的子树中的最大值,要么是除去以u为根的树外的剩余结点数dp[u]=max(max(cnt[v]),n-cnt[u]);

 

3、AC代码:

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;#define N 20005#define INF 20010int v[N],next[N],head[N];int tot,ans,res,n;int cnt[N];int dp[N];void add_edge(int a,int b){    v[tot]=b;    next[tot]=head[a];    head[a]=tot++;}void dfs(int u,int fa){    cnt[u]=1;    for(int i=head[u];i!=-1;i=next[i])    {        int vv=v[i];        if(vv!=fa)        {            dfs(vv,u);            cnt[u]+=cnt[vv];            ans=max(ans,cnt[vv]);        }    }    dp[u]=max(ans,n-cnt[u]);    res=min(res,dp[u]);}int main(){    int a,b;    scanf("%d",&n);    memset(head,-1,sizeof(head));    memset(cnt,0,sizeof(cnt));    tot=0;    for(int i=1;i<n;i++)    {        scanf("%d%d",&a,&b);        add_edge(a,b);        add_edge(b,a);    }    ans=-INF;    res=INF;    dfs(1,0);    if(res>n/2)    printf("NONE\n");    else    {        for(int i=1;i<=n;i++)        {            if(dp[i]<=n/2)            printf("%d\n",i);        }    }    return 0;}


 

0 0
原创粉丝点击