poj 3107 Godfather

和poj 1655类似，也是拆了一个树求什么值。这题我改进了一下，把两个DFS合一了。

#include <stdio.h>
#define N 50005
int fst[N],nxt[N<<1],to[N<<1];
int n,s[N],w[N],min;
int dfs(int u,int fa)
{
    int i,v,t=n-1;
    s[u] = 1; w[u] = 0;
    for(i = fst[u]; i != 0; i = nxt[i])
    {
        v = to[i];
        if(v != fa)
            s[u] += dfs(v,u);
    }
    //printf("s[%d]:%d\n",u,s[u]);
    for(i = fst[u]; i != 0; i = nxt[i])
    {
        v = to[i];
        if(v != fa)
        {
            t -= s[v];
            if(s[v] > w[u]) w[u]=s[v];
        }
    }
    if(t > w[u]) w[u]=t;
    //printf("w[%d]:%d\n",u,w[u]);
    if(w[u] < min) min = w[u];
    return s[u];
}
int main()
{
    int a,b,i,j,f=1;
    scanf("%d",&n);
    min = n;
    for(i = j = 1; j < n; j++)
    {
        scanf("%d%d",&a,&b);
        to[i] = b;
        nxt[i] = fst[a];
        fst[a] = i++;
        to[i] = a;
        nxt[i] = fst[b];
        fst[b] = i++;
    }
    dfs(1,0);
    for(i = 1; i <= n; i++)if(w[i] == min)
    {
        if(f) printf("%d",i), f = 0;
        else printf(" %d",i);
    }
    printf("\n");
    return 0;
}