还记得小时候经常看少儿频道,“智慧树下你和我,智慧树下欢乐多,智慧树下做游戏,欢乐多又多”,长大后智慧树也没放过我;
zzy今天来到的智慧树下摘果子,智慧树很生气所以他想惩罚zzy,他就让树上的果子全部砸到zzy的头上。这棵树就是一个树形结构,每个节点有一颗果子。zzy想被砸的最轻,所以他可以选择一个节点为根,使得以这个点为根的树,所有点的深度之和最大。(这样果子离他的头就近了)。
Input
输入包含一个数字N,代表有N个点.N<=1000000 下面N-1条边.
Output
输出zzy找到的点,如果具有多个解,请输出编号最小的那个.
Sample Input Copy
8
1 4
5 6
4 5
6 7
6 8
2 4
3 4
Sample Output Copy
7
题意:一棵n个节点的树,找出一个点,使得以这个点为根时,所有点的深度之和最大(如果最大值相同,输出拥有更小编号的点 )。
建树我建议用vector,虽然我用的还不是很熟练,但我觉得是真的好用,我看很多博主用的都是结构体,感觉写出来代码有点乱,初学者很容易搞蒙,比如说我;然后遍历每一个节点,方程呢就是a[kk]=a[x]+(n-2*siz[kk]);
注意建树是先每层一个到最后一个,然后再返回的;
代码:
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;
int siz[1000010];//节点个数
int dp[1000010];//深度
int n;
int a[1000010];
vector<vector<int> >f(1000010);
void dfs(int x,int fa)//建树
{
siz[x]=1;
for(int i=0;i<f[x].size();i++)
{
int kk = f[x][i];
if(kk==fa)
continue;
dp[kk]=dp[x]+1;
dfs(kk,x);
siz[x]+=siz[kk];
}
}
void ddfs(int x,int fa)//遍历每一个结点
{
for(int i=0;i<f[x].size();i++)
{
int kk=f[x][i];
if(kk==fa)
continue;
a[kk]=a[x]+(n-2*siz[kk]);///kk这个结点的最大深度和
ddfs(kk,x);
}
return ;
}
int main()
{
while(~scanf("%d",&n))
{
int ans =0;
for (int i=1;i<=n;i++)
f[i].clear();
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
f[y].push_back(x);
f[x].push_back(y);
}
dfs(1,0);
ddfs(1,0);
a[1]=1;
for(int i=2;i<=n;i++)
{
if(a[ans]<a[i])
ans=i;
}
printf("%d\n",ans);
}
return 0;
}