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 ;
}