POJ 3107-Godfather（树形dp）

题意：

有n个节点的树，删除一个点，得到的最大联通分支最小，求这样点的集合

分析：

dp[i]表示删除i所得最大联通分支，遍历一遍节点即可，该题用vector会超时

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define N 50010
#define read freopen("in.txt", "r", stdin)
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod =  1000000007;
int dp[N],n,num[N],used[N],ans[N],len,head[N];
struct edge{
    int t,next;
}e[N*2];
int max(int a,int b){
    return a>b?a:b;
}
void add(int a,int b){
    e[len].t=b;
    e[len].next=head[a];
    head[a]=len++;
}
int dfs(int root){
    num[root]=1;
    used[root]=1;
    for(int i=head[root];i!=-1;i=e[i].next){
        int son=e[i].t;
        if(used[son])continue;
        num[root]+=dfs(son);
        dp[root]=max(dp[root],num[son]);
    }
    dp[root]=max(dp[root],n-num[root]);
    return num[root];
}
int main()
{
    while(~scanf("%d",&n)){
        memset(dp,0,sizeof(dp));
        memset(used,0,sizeof(used));
        memset(head,-1,sizeof(head));
        int a,b;
        len=0;
        for(int i=0;i<n-1;++i){
            scanf("%d%d",&a,&b);
            add(a,b);
            add(b,a);
        }
        dfs(1);
        int l,minv=INF;
        for(int i=1;i<=n;++i){
            if(dp[i]<minv){
                l=0;
                ans[l++]=i;
                minv=dp[i];
            }
            else if(dp[i]==minv)
                ans[l++]=i;
        }
        for(int i=0;i<l;++i){
            printf("%d ",ans[i]);
        }
        printf("\n");
    }
return 0;
}