【bzoj 1131】 Sta 【POI2008】

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求出以1为根时的答案，拓展出以其他节点为根的答案即可，下面是程序：

#include<stdio.h>
#include<iostream>
#define ll long long
using namespace std;
const int N=1000005;
struct edge{
	int v,next;
}e[N<<1];
int head[N],k,n;
ll f[N],sum[N];
void read(int &s){
	s=0;
	char c=getchar();
	while(c<'0'||c>'9'){
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		s=s*10+c-'0';
		c=getchar();
	}
}
void add(int u,int v){
	e[++k]=(edge){v,head[u]};
	head[u]=k;
}
void dfs(int u,int fa,int h){
	int i;
	sum[u]=1;
	f[u]=h;
	for(i=head[u];i;i=e[i].next){
		if(e[i].v!=fa){
			dfs(e[i].v,u,h+1);
			f[u]+=f[e[i].v];
			sum[u]+=sum[e[i].v];
		}
	}
}
void dp(int u,int fa){
	int i;
	for(i=head[u];i;i=e[i].next){
		if(e[i].v!=fa){
			f[e[i].v]=f[u]+n-2*sum[e[i].v];
			dp(e[i].v,u);
		}
	}
}
int main(){
	int i,u,v,s=1;
	read(n);
	for(i=1;i<n;i++){
		read(u);
		read(v);
		add(u,v);
		add(v,u);
	}
	dfs(1,0,1);
	dp(1,0);
	for(i=2;i<=n;i++){
		if(f[s]<f[i]){
			s=i;
		}
	}
	printf("%d\n",s);
	return 0;
}