POJ 2378 Tree Cutting

Tree Cutting

给一棵n个节点的无根树，问去掉哪个节点后可以使所有连通分支的节点数均<=n/2。很简单的树形DP，dp[i]表示去掉该节点后连通分支中最多的节点个数。然后dfs一遍，找出所有点的dp值，再扫一遍，遇到小于等于n/2的，直接输出。。。

#include <cstdio>
#include <cstring>
#define maxn 10001
int dp[maxn] ;
int first[maxn] ;
int next[maxn<<1] ;
int edge[maxn<<1] ;
int n ;

inline void add(int p , int q , int e){
    next[e]  = first[p] ;
    first[p] = e ;
    edge[e]  = q ;
}

int dfs(int v , int f){
    int cnt = 1 ;
    int tmp ;

    for(int i = first[v] ; i != -1 ; i = next[i]){
        int u = edge[i] ;
        if(u == f)
            continue ;
        tmp = dfs(u , v) ;
        dp[v] = tmp > dp[v] ? tmp : dp[v] ;
        cnt += tmp ;
    }
    dp[v] = (n - cnt) > dp[v] ? (n - cnt) : dp[v] ;
    return cnt ;
}

int main(){
    scanf("%d" , &n) ;
    int p , q ;
    memset(next , -1 , sizeof(next))  ;
    memset(first, -1 , sizeof(first)) ;
    for(int i = 1 ; i < n ; i ++){
        scanf("%d%d" , &p , &q) ;
        add(p , q , i<<1) ;
        add(q , p , i<<1|1) ;
    }
    dfs(1 , 0) ;
    bool flag = 0 ;
    for(int i = 1 ; i <= n ; i ++){
        if(dp[i] <= n/2)
            printf("%d\n" , i) , flag = 1 ;
    }
    if( !flag )
        puts("NONE") ;
    return 0 ;
}