poj 2378 Tree Cutting (树形dp)

标签：

Description

Input

Output

Sample Input

10
1 2
2 3
3 4
4 5
6 7
7 8
8 9
9 10
3 8

Sample Output

3
8

Hint

Source

类似于poj3107

求删除一个点后，形成的森林中，每棵树中最大节点数不大于原树节点数目的一半。

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define ll long long
#define N 11000
#define mem(a,t) memset(a,t,sizeof(a))
const int inf=1000005;
int cnt,n;
struct node
{
    int v,next;
}e[N*2];
int head[N];
int num[N];
int ans[N];
int n1;
void add(int u,int v)
{
    e[cnt].v=v;
    e[cnt].next=head[u];
    head[u]=cnt++;
}

void dfs(int u,int len,int fa)
{
    int i,v,sum=0,tmp=0;
    num[u]=1;
    for(i=head[u];i!=-1;i=e[i].next)
    {
        v=e[i].v;
        if(v!=fa)
        {
            dfs(v,len+1,u);
            tmp=max(tmp,num[v]);
            num[u]+=num[v];
        }
    }

    int t=max(tmp,n-num[u]);
    if(t<=n/2)
        ans[n1++]=u;
}

int main()
{
    //freopen("in.txt","r",stdin);
    int i,u,v;
    while(~scanf("%d",&n))
    {
        cnt=0;
        mem(head,-1);
        for(i=1;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        mem(num,0);
        n1=0;
        dfs(1,0,-1);
        sort(ans,ans+n1);
        for(i=0;i<n1;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}

poj 2378 Tree Cutting (树形dp)

标签：

原文地址：http://blog.csdn.net/u011721440/article/details/45392281