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 long

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