POJ 3107 Godfather

Godfather

樹形DP簡單題，就是找到這樣的點：刪除該點後最大的連通子樹所包含的節點個數最小。然後將這些點依次輸出。dp[i] = max{dp[j] , n-number} , j是i的兒子節點，number是i所在子樹的總結點個數。

#include <cstdio>
#include <cstring>
#include <vector>

#define maxn 50005
using std::vector ;
int n ;
int result ;
int dp[maxn] ;
vector<int>G[maxn] ;

int dfs(int v  , int f){
    int ans = 1 ;
    for(int i = 0 ; i != G[v].size() ; i ++){
        int u = G[v][i] ;
        if(u == f){
            continue ;
        }
        int tmp = dfs(u , v) ;
        ans += tmp ;
        if(dp[v] < tmp)
            dp[v] = tmp ;
    }
    if(n - ans > dp[v]){
        dp[v] = n - ans ;
    }
    result = result < dp[v] ? result : dp[v] ;
    return ans ;
}
int main(){
    scanf("%d" , &n) ;
    int a , b ;
    for(int i = 1 ; i < n ;  i ++){
        scanf("%d%d" ,&a , &b) ;
        G[a].push_back(b) ;
        G[b].push_back(a) ;
    }
    result = 50005 ;
    dfs(1 , 0) ;
    int flag = 0 ;
    for(int i = 1 ; i <= n ; i ++){
        if(dp[i] == result){
            if(flag)
                putchar(' ') ;
            printf("%d" , i) ;
            flag = 1 ;
        }
    }
    printf("\n") ;
    return 0 ;
}