BZOJ 1131: [POI2008]Sta

1131: [POI2008]Sta

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1370  Solved: 486
[Submit][Status][Discuss]

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

HINT

 

Source

 

[Submit][Status][Discuss]

分析

一开始不妨拿1当作根节点，预处理出每个子树的结点个数，然后从1开始DFS，在相邻点种枚举新的根，可以O(1)的得到新的SON数组以及深度之和。

代码

#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
 
using namespace std;
 
const int N = 5000005;
 
int n, id = 1;
 
long long sum, ans, son[N];
 
int hd[N], nt[N], to[N], tot;
 
void add(int x, int y)
{
    nt[tot] = hd[x]; to[tot] = y; hd[x] = tot++;
    nt[tot] = hd[y]; to[tot] = x; hd[y] = tot++;
}
 
void prework(int u, int f, int d)
{
    sum += d; son[u] = 1;
     
    for (int i = hd[u]; ~i; i = nt[i])
        if (to[i] != f)prework(to[i], u, d + 1), son[u] += son[to[i]];
}
 
void search(int u, int f, long long s)
{
    if (ans < s || (ans == s && id > u))
        ans = s, id = u;
     
    for (int i = hd[u]; ~i; i = nt[i])
        if (to[i] != f)search(to[i], u, s + n - son[to[i]]*2);
}
 
signed main(void)
{
    scanf("%d", &n); 
     
    memset(hd, -1, sizeof(hd)); tot =0; 
     
    for (int i = 1, x, y; i < n; ++i)
        scanf("%d%d", &x, &y), add(x, y);
         
    prework(1, -1, 0); search(1, -1, sum);
     
    printf("%d
", id);
}

@Author: YouSiki