592D - Super M

最新推荐文章于 2022-05-09 17:38:21 发布

wamach

最新推荐文章于 2022-05-09 17:38:21 发布

阅读量310

点赞数

CC 4.0 BY-SA版权

分类专栏： CodeForces

本文链接：https://blog.youkuaiyun.com/wamach/article/details/74894194

CodeForces 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种解决特定树形结构中最短路径问题的算法。该算法应用于一棵n个点的树上，给定m个点，目标是找到从任意点出发经过这m个点的最短路径长度及最佳起点编号。通过重构一棵以这m个点为叶子节点的树，并利用广度优先搜索(BFS)与深度优先搜索(DFS)技术，实现了路径总长的有效计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在一棵n个点的树上给定m个点，求从树上任意点出发，到任意点，经过这m个点所经过的路程最小值，和此时最小的起点编号。

重建一颗以这m个点为叶子结点的树，答案即这棵树的路径总长*2-直径长

m等于1的时候我特判自爆了，起点编号一直不对，WA了一个多小时QAQ

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define rep(i,j,k) for(i=j;i<=k;++i)
#define per(i,j,k) for(i=j;i>=k;--i)
#define LL long long 
const int N=130000;
int n,m;bool flg[N];
int he[N],ne[N<<1],to[N<<1],tot;
int rt,q[N],d[N],s;bool vis[N];
void add(int x,int y){
	to[++tot]=y;ne[tot]=he[x];he[x]=tot;
}
void BFS(){
	int Ft=0,Rr=1,i,u,v;
	rep(i,1,n)vis[i]=!flg[i];
	for(vis[q[0]=rt]=1;Ft!=Rr;++Ft)
		for(i=he[u=q[Ft]];i;i=ne[i])if(!vis[v=to[i]]){
			vis[q[Rr++]=v]=1;d[v]=d[u]+1;
		}
}
void DFS(int x,int e){
	int i,y;
	for(i=he[x];i;i=ne[i])if(i!=e){
		DFS(y=to[i],i^1);
		if(flg[y])flg[x]=1,++s;
	}
}
int main(){
	int i,x,y;
	scanf("%d%d",&n,&m);tot=1;
	rep(i,2,n){
		scanf("%d%d",&x,&y);
		add(x,y),add(y,x);
	}
	if(m==1){
		scanf("%d",&rt);printf("%d\n0\n",rt);return 0;
	}
	for(rt=n+1;m--;){
		scanf("%d",&x);
		flg[x]=1;
		rt=min(x,rt);
	}
	DFS(rt,0);BFS();
	x=1;rep(i,2,n)if(d[i]>d[x])x=i;rt=x;
	memset(d,0,sizeof d);BFS();
	x=1;rep(i,2,n)if(d[i]>d[x])x=i;
	printf("%d\n%d\n",min(rt,x),(s<<1)-d[x]);
	return 0;
}