POJ 2378 Tree Cutting

题目大意

给定一个无向图形成一棵树，截去某个点后，森林中的树要是每棵树中的节点数都不超过原来总结点数的1/2，那么就把这个点输出

否则输出NONE

前面也做了思想一样的题目POJ 1655 Balancing Act && POJ 3107 Godfather

不理解可以看看

/*
sum[i]表示i对应子树中的节点总数
down[i]表示截去i后，i下方子树断开成森林中最大的那棵子树的节点数
i取走后，往上走的子树长度为 n -sum[i]
rec[i] = max(down[i] , n -sum[i]);
*/
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int N = 10005;
int first[N] , k , down[N] , sum[N] , rec[N];

struct Edge{
    int y , next;
}e[N<<1];

void add_edge(int x , int y)
{
    e[k].y = y , e[k].next = first[x];
    first[x] = k++;
}

void  dfs1(int u , int fa)
{
    sum[u] = 1;
    for(int i = first[u] ; i!=-1 ; i=e[i].next)
    {
        int v = e[i].y;
        if(v == fa) continue;
        dfs1(v , u);
        sum[u] += sum[v];
    }
}

void dfs2(int u , int fa)
{
    for(int i = first[u] ; i!=-1 ; i=e[i].next)
    {
        int v = e[i].y;
        if(v == fa) continue;
        dfs2(v , u);
        down[u] = max(sum[v] , down[u]);
    }
}

int main()
{
   // freopen("a.in" , "r" , stdin);
    int n , x , y;
    while(scanf("%d" , &n)==1){
        memset(first , -1 , sizeof(first));
        k=0;
        for(int i=1 ; i<n ; i++){
            scanf("%d%d" , &x , &y);
            add_edge(x , y);
            add_edge(y , x);
        }

        memset(down , 0 , sizeof(down));
        dfs1(1 , -1);
        dfs2(1 , -1);
        int num=0;
        for(int i=1 ; i<=n ; i++){
            rec[i] = max(down[i] , n-sum[i]);
            if(rec[i]<=n/2){
                printf("%d
" , i);
                num++;
            }
        }
        if(num == 0) puts("NONE");
    }
    return 0;
}