根 ssl 2668 优美的暴力

题目大意

一棵n个节点的树，找出一个点，使得以这个点为根时，所有点的深度之和最大（如果最大值相同，输出拥有更小编号的点 ）。

分析

先用dfs求出以1为根时所有节点的深度和和每个子树的节点数low，注意到可以O(1)换根，所以可以枚举每个点计算贡献即可。
公式为dep=dep-low[i]+(n-low[i]);

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#include<algorithm>

using namespace std;

struct arr{
    int x,y,next;
}edge[3000000];

int ls[3000000];
int low[3000000];
int max_,ans,dep,n;
int edge_m;

void add(int x,int y)
{
    edge[++edge_m]=(arr){x,y,ls[x]},ls[x]=edge_m;
    edge[++edge_m]=(arr){y,x,ls[y]},ls[y]=edge_m;
}

void dfs(int x,int r,int d)
{
    dep+=d;
    low[x]=1;
    for (int i=ls[x];i!=0;i=edge[i].next)
    {
        if (edge[i].y==r) continue;
        dfs(edge[i].y,x,d+1);
        low[x]+=low[edge[i].y];
    }
}

void answer(int x,int r,int dep)
{
    if (x!=1)
    {
        dep=dep-low[x]+(n-low[x]);
        if ((dep>max_)||((dep==max_)&&(ans>x)))
            max_=dep,ans=x;
    }
    for (int i=ls[x];i!=0;i=edge[i].next)
    {
        if (edge[i].y==r) continue;
        answer(edge[i].y,x,dep);
    }
}
int main()
{
    edge_m=1;
    scanf("%d",&n);
    for (int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    dfs(1,0,0);
    max_=dep;
    ans=1;
    answer(1,0,dep);
    printf("%d",ans);
}