bzoj1131 树形dp

闲着无聊做水题2
先求出任意一个为根的深度和,深度,子树大小等,然后在第二次进行转移,然后深搜就可以了

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e6+10;
typedef long long ll;
ll f[N];int dep[N],s[N],n,len=0,last[N],ans;
struct Edge{int to,next;Edge(int to=0,int next=0):to(to),next(next){}}e[N<<2];
void add_edge(int u,int v){e[++len]=Edge(v,last[u]);last[u]=len;}
void dfs1(int x,int fa)
{
    s[x]=1;f[x]=dep[x];
    for(int i=last[x];i;i=e[i].next){
        int id=e[i].to;
        if(id==fa)continue;
        dep[id]=dep[x]+1;
        dfs1(id,x);
        f[x]+=f[id];s[x]+=s[id];
    }
}
void dfs2(int x,int fa)
{
    for(int i=last[x];i;i=e[i].next)
        if(e[i].to!=fa)
        {
            f[e[i].to]=f[x]+n-2*s[e[i].to];
            dfs2(e[i].to,x);    
        }
}
int main()
{
    scanf("%d",&n);
    for(int x,y,i=1;i<n;i++) {
        scanf("%d%d",&x,&y);
        add_edge(x,y);add_edge(y,x);
    }
    dfs1(1,0);dfs2(1,0);
    for(int i=1;i<=n;i++) if(f[i]>f[ans])ans=i;
    cout<<ans<<endl;
    return 0;
}