Atcoder ABC D题题解

题目链接

题目要求我们寻找一个包含所有指定顶点的子树，且这个子树的顶点数量最少。为了方便叙述，记$A_{i}$表示第i个被要求包含的顶点

因为树上两个顶点之间的路径是唯一的，我们可以以$A_{1}$为根节点，用dfs求出$A_{1}$到$A_{2},A_{3},A_{4}......A_{K}$的路径，然后把经过的点都标记上，最后统计有多少个点被标记，就是所求最小子树的大小。

但很明显，上述思路中会有一些顶点被重复访问和标记，考虑优化。

我们可以转变思路，分别求从$A_{2},A_{3},A_{4}...A_{K}$开始dfs求其到$A_{1}$路径，并标记经过的顶点，如果访问到一个顶点$u$被标记过，就可以停止dfs，开始回溯并标记走过的路。这样就可以杜绝重复的访问，提高效率。

给出代码

#include<bits/stdc++.h>
using namespace std;
int head[200010],pre[200010],N,K,cnt,ans;
bool szhash[200010];
struct edge
{
	int next;
	int to;
}Edge[400020];

void add(int u,int v)
{
	Edge[++cnt].next = head[u];Edge[cnt].to = v;head[u] = cnt;
}

void Set(int u)
{
	if(u) {szhash[u] = 1;/**printf("recommend %d\n",u)***/;Set(pre[u]);}
}

void dfs(int fa,int u)
{
	pre[u] = fa;
	//printf("go to %d\n",u);
	if(szhash[u]) {Set(u);return;}
	for(int i = head[u];i;i = Edge[i].next)
	{
		if(fa == Edge[i].to) continue;
		dfs(u,Edge[i].to);
	}
}

int main()
{
	int u,v;
	scanf("%d %d",&N,&K);
	for(int i = 1;i<N;i++){scanf("%d %d",&u,&v);add(u,v);add(v,u);}
	scanf("%d",&u);szhash[u] = 1;
	for(int i = 1;i<K;i++){scanf("%d",&u);if(!szhash[u]){dfs(0,u);}}
	for(int i = 1;i<=N;i++) if(szhash[i]) ans++;
	printf("%d",ans);
	return 0;
}