#A. 八中开会
- 传统题1000ms256MiB
八中有N个教室,它们之间由N-1条边连接而成。
现在校长希望在某个教室召集班长们开会,那么这个地方应该比较方便才好,也就是说各位班长到这个会议点,所走过的距离的总和应该越小越好。
若有多个节点都满足条件,则选择节点编号最小的那个教室。
输入
第一行给出数字N,N小于等于50000
接下来N行,给出数字A,B,表示A教室与B教室之间有一条边
输出
输出应该在哪个教室开会及走过的距离总和最小值为多少
输入数据 1
4
1 2
2 3
3 4
Copy
输出数据 1
2 4
先看一个样例
可以看出,在pn点集合的代价可以继承其父亲点p的代价,减去pn及其子树数,加上剩余点数。
#include<bits/stdc++.h>
using namespace std;
int n,x,y,ans[50005],num[50005],ansd,ansp;
vector<int>d[50005];
//num[i]--i及其子树数
//ans[i]--i为集合点的代价
void trc(int p,int f,int dep)//ans[1]和num[i]
{
for(int i=0;i<=int(d[p].size()-1);i++)
if(d[p][i]!=f)
{
int pn=d[p][i];
ans[1]+=dep+1;
trc(pn,p,dep+1);
num[p]+=num[pn];
}
return ;
}
void forbetter(int p,int f)//遍历树得到ans[i]
{
for(int i=0;i<=int(d[p].size()-1);i++)
if(d[p][i]!=f)
{
int pn=d[p][i];
ans[pn]=ans[p]+n-2*num[pn];
forbetter(pn,p);
}
return ;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n-1;i++)
{
scanf("%d%d",&x,&y);
d[x].push_back(y);
d[y].push_back(x);
num[i]=1;
}
num[n]=1;
trc(1,0,0);
ansd=ans[1];
ansp=1;
forbetter(1,0);
for(int i=1;i<=n;i++)
if(ans[i]<ansd)
{
ansd=ans[i];
ansp=i;
}
printf("%d %d",ansp,ansd);
return 0;
}