bzoj1131 [POI2008]Sta 树形DP

题意：给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大

由于结论题做多了第一反应结论。
觉得找直径好像没什么问题，但是WA。并找不出为什么错= =。后来发算就算不wa也会tle，因为大一点的菊花图我就炸了，因为我找不出最小的，如果要找出所有的直径端点，如果是菊花图，复杂度就是n^2..
正解其实挺简单啦，明显的dp，每次换根的贡献O(1)计算，注意在一个点上方的贡献固定，不用计算。
算是吸取个教训咯。。这种题目居然没有一眼切。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e6+5;
int n,m,dep[N],ans;
int head[N],next[N],go[N],tot,size[N],f[N];
inline void add(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
inline void dfs1(int x,int fa)
{
    size[x]=1;
    dep[x]=dep[fa]+1;
    f[x]=dep[x];
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (v!=fa)
        {
            dfs1(v,x);
            size[x]+=size[v];
            f[x]+=f[v];
        }
    }
}
inline void dfs2(int x,int fa)
{
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (v!=fa)
        {
            f[v]=f[x]+n-2*size[v];
            dfs2(v,x);
        }
    }
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n-1)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs1(1,0);
    dfs2(1,0);
    fo(i,1,n)
    if (f[i]>f[ans])ans=i;
    printf("%d\n",ans);
}