根

2017年8月7日提高组T1 根

Time Limit:20000MS Memory Limit:262144K
Total Submit:50 Accepted:20
Case Time Limit:2000MS

Description

一天，小B得到了一棵n个节点的树。无聊至极的小B想要找出一个点，使得以这个点为根时，所有点的深度之和最大。但小B打开手机，发现他最爱的re:creator又更新啦，于是这个无聊的任务就交给你了。

Input

第一行一个数n，接下来n-1行，每行两个数，表示一条边。

Output

一行一个整数，表示所有点深度之和的最大值的那个点。
如果最大值相同，输出拥有更小编号的点

Sample Input

5
2 1
3 1
4 2
5 3
Sample Output

4
Hint

对于前30%的数据，n<=1000.
对于前100%的数据，n<=1000000.
做法：先求出以1为根时所有节点的深度和，注意到可以O(1)换根，所以可以枚举每个点计算贡献即可。
代码如下：

#include <cstdio>#include <cstring>using namespace std;struct arr{    int to,next;}f[2000007];int ls[2000007],list[4000007],n,d[1000007],ans=0,zs[1000007],fa[2000007];bool v[1000007];void spfa(int x){    for (int i=1;i<=n;i++)        d[i]=0x7f7f7f7f;    d[x]=1;    int head=0,tail=1;    list[1]=x;    v[x]=1;    while (head<tail)    {        head++;        int t=ls[list[head]];        while (t>0)        {            if (d[list[head]]+1<d[f[t].to])            {                d[f[t].to]=d[list[head]]+1;                if (!v[f[t].to])                {                    v[f[t].to]=1;                    tail++;                    list[tail]=f[t].to;                }            }            t=f[t].next;        }        v[list[head]]=0;    }}int dfs(int x){    v[x]=true;    for (int i=ls[x];i;i=f[i].next)        if (!v[f[i].to])        {            zs[x]+=dfs(f[i].to)+zs[f[i].to];        }    return 1;}int main(){    //freopen("root1.in","r",stdin);    scanf("%d",&n);    int x,y,e=0;    for (int i=1;i<=n-1;i++)    {        scanf("%d%d",&x,&y);        f[++e].to=y;        f[e].next=ls[x];        ls[x]=e;        f[++e].to=x;        f[e].next=ls[y];        ls[y]=e;    }    int ma=0,ans2=0,answer=3000000;    spfa(1);    for (int i=1;i<=n;i++)        ans+=d[i];    dfs(1);    for (int i=1;i<=n;i++)        zs[i]++;    memset(list,0,sizeof(list));    memset(v,1,sizeof(v));    int head=0,tail=1;    list[1]=1;    v[1]=0;    fa[1]=ans;    while (head<tail)    {        head++;        for (int i=ls[list[head]];i;i=f[i].next)            if (v[f[i].to])            {                v[f[i].to]=0;                ans2=fa[head]-zs[f[i].to]-zs[f[i].to]+n;                tail++;                fa[tail]=ans2;                list[tail]=f[i].to;                 if (ans2>ma || (ans2==ma && answer>f[i].to))                {                    ma=ans2;                    answer=f[i].to;                }                               }    }    printf("%d",answer);    fclose(stdin);}