题目要求我们寻找一个包含所有指定顶点的子树,且这个子树的顶点数量最少。为了方便叙述,记$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;
}