【BZOJ】 【1131】【Sta】 【POI 2008】

一大早起来A了一道水题瞬间神清气爽

1131: [POI2008]Sta

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 1686   Solved: 634
[ 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] 

HOME Back

这道题只需要了解树上任意两点只会有一条简单路径，所以每次经过一条边，就会将一个子树深度加一，另一个子树深度减一

两次dfs 
dfs0：  
求出 size(i) 和 deep(i)  
dfs1：  
dp(v)=dp(u)+n−2∗size(v)

代码如下：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 1000001
using namespace std;
int n , ans , poi ;
long long dp[M] , son[M] , dip[M];
int ind[M] , e[M * 2] , nex[M * 2] , cnt;
void add(int a , int b){
	nex[++cnt] = ind[a];
	e[ind[a] = cnt] = b;
}
int dfs(int v , int f){
	dip[v] = dip[f] + 1;
	son[v] = 1;
	for(int i = ind[v] ; i ; i = nex[i])
		if(e[i] != f)
			son[v] += dfs(e[i] , v);
	return son[v];
}

void dfs_(int v , int f){
	if(dip[v] == 1)
		for(int i = 1 ; i <= n ; ++i)
			dp[v] += dip[i];
	else
		dp[v] = dp[f] + n - 2 * son[v]; 
	for(int i = ind[v] ; i ; i = nex[i])
		if(e[i] != f)
			dfs_(e[i] , v);
}
int main(){
	scanf("%d",&n);
	for(int i = 1 ; i < n ; ++i){
		int a , b;
		scanf("%d%d",&a,&b);
		add(a , b);
		add(b , a);
	}
	dfs(1 , 0);
	dfs_(1 , 0);
	for(int i = 1; i <= n ; ++i)
		if(dp[i] > dp[poi])poi = i;
	printf("%d\n",poi);
}