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