bzoj1131[POI2008]Sta*

bzoj1131[POI2008]Sta

题意：

给出一个n个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大。n≤1000000。

题解：

两次dfs。第一次dfs维护子树的大小、节点的深度、以子树的根为根的子树深度和。第二次dfs维护以某节点为根除了以1为根时它的子树之外的所有节点的深度和。最后比较所有节点两个深度和相加的值。具体看代码。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define maxn 1000010
#define ll long long
using namespace std;

inline int read(){
    char ch=getchar(); int f=1,x=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return f*x;
}
struct e{int t,n;}es[maxn*2]; int g[maxn],ess;
void pe(int f,int t){es[++ess]=(e){t,g[f]}; g[f]=ess;}
ll ds[maxn],sz[maxn],fads[maxn],dep[maxn]; int n,ans,fa[maxn];
void dfs1(int x){
    sz[x]=1; ds[x]=0;
    for(int i=g[x];i;i=es[i].n)if(es[i].t!=fa[x]){
        fa[es[i].t]=x; dep[es[i].t]=dep[x]+1; dfs1(es[i].t);
        sz[x]+=sz[es[i].t]; ds[x]+=ds[es[i].t]+sz[es[i].t];
    }
}
void dfs2(int x,ll a){
    fads[x]=a+n-sz[x]; ll sm=fads[x];
    for(int i=g[x];i;i=es[i].n)if(es[i].t!=fa[x])sm+=ds[es[i].t]+sz[es[i].t];
    for(int i=g[x];i;i=es[i].n)if(es[i].t!=fa[x])dfs2(es[i].t,sm-(ds[es[i].t]+sz[es[i].t]));
}
int main(){
    n=read(); inc(i,1,n-1){int x=read(),y=read(); pe(x,y); pe(y,x);} dfs1(1); dfs2(1,0); 
    ans=0; inc(i,1,n)if(fads[i]+ds[i]>fads[ans]+ds[ans])ans=i; printf("%d",ans); return 0;
}

20160919