BZOJ 1131 [POI 2008] 解题报告

1131: [POI2008]Sta

Description

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

Input

给出一个数字N,代表有N个点.N<=1000000 下面N-1条边.

Output

输出你所找到的点,如果具有多个解,请输出编号最小的那个.

Sample Input

8
1 4
5 6
4 5
6 7
6 8
2 4
3 4

Sample Output

7

【解题报告】
两次dfs
dfs0：
求出size(i)和deep(i)
dfs1：
dp(v)=dp(u)+n−2∗size(v)

代码如下：

/**************************************************************    Problem: 1131    User: onepointo    Language: C++    Result: Accepted    Time:7896 ms    Memory:61296 kb****************************************************************/#include<cstdio>#include<cstring>#include<algorithm>using namespace std;#define N 1000010#define LL long longint n,cnt,head[N],ans;struct Edge{int to,nxt;}e[N<<1];int size[N],deep[N];LL dp[N];void adde(int u,int v){    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;    e[++cnt].to=u;e[cnt].nxt=head[v];head[v]=cnt;}void dfs0(int u,int fa){    size[u]=1;dp[u]=(LL)deep[u];    for(int i=head[u];~i;i=e[i].nxt)    {        int v=e[i].to;        if(v==fa) continue;        deep[v]=deep[u]+1;        dfs0(v,u);        dp[u]+=dp[v];        size[u]+=size[v];    }}void dfs1(int u,int fa){    for(int i=head[u];~i;i=e[i].nxt)    {        int v=e[i].to;        if(v==fa) continue;        dp[v]=dp[u]+n-2*size[v];        dfs1(v,u);    }}int main(){    cnt=-1;    memset(head,-1,sizeof(head));    scanf("%d",&n);    for(int i=1;i<n;++i)    {        int u,v;scanf("%d%d",&u,&v);        adde(u,v);    }    dfs0(1,1);    dfs1(1,1);    for(int i=1;i<=n;++i)    if(dp[i]>dp[ans]) ans=i;    printf("%d\n",ans);    return 0;}