看了这个题后,第一反应就是爆搜。
怎么搜呢,很简单,直接从所有 v [ i ] v[i] v[i] 的 lca 开始 DFS ,找到了需要的 v [ i ] v[i] v[i] 就直接回溯,把一路上的点打上标记,最后统计标记,得到 a n s ans ans 并输出。
以此题第一个样例为例,需要保留 1 1 1 , 3 3 3 , 5 5 5 ,它们的 lca 为 1 1 1 ,直接从 1 1 1 开始 DFS ,搜到 5 5 5 后把 1 1 1 , 2 2 2 , 5 5 5 标记上,搜到 3 3 3 就把 1 1 1 , 3 3 3 标记上, 1 1 1 就直接标记 1 1 1 即可,代码实现如下
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <queue>
#define INF 1000010
using namespace std;
queue<int> q;
int n,k,v[INF],t;
int f[INF][40],cnt,d[INF];
//以上多为模板/题中所需,没必要解释了
bool vis[INF],need[INF],viss[INF],ff;
//vis数组记录DFS途中已经到达的点,用于dfs剪枝,这个应该都知道吧
//need用于记录哪个点需要保留
//viss用于记录最后树上需要保留的点
//举个例子,对于图中有viss[1]=viss[2]=viss[3]=viss[5]=1
//ff在dfs中记录 是否在某个子树找到了 需要保留的v[i]节点
struct Tu{
int next,to;
}map[INF];
int head[INF];
//链式前向星存图
void add(int u,int v);
void DFS(int d);
void BFS();
int LCA(int x,int y);
int main()
{
scanf("%d%d",&n,&k);
t=(int)(log(n)/log(2))+1;
for(int i=1;i<n;++i)
{
int a,b;
scanf("%d%d",&a,&b);
add(a,b);add(b,a);
}
for(int i=1;i<=k;++i)
scanf("%d",&v[i]),need[v[i]]=1;//使用need数组可以将后期查找的复杂度降为O(1)
BFS();
int llca=v[1];
//求出所有v[i]的lca,并从lca开始DFS
for(int i=2;i<=k;++i)
llca=LCA(llca,v[i]);
DFS(llca);
//统计答案
int ans=0;
for(int i=1;i<=n;++i)
if(viss[i]) ++ans;
printf("%d\n",ans);
return 0;
}
void add(int u,int v)//链式前向星存图
{
++cnt;
map[cnt].to=v;
map[cnt].next=head[u];
head[u]=cnt;
}
void DFS(int d)
{
vis[d]=1;
if(need[d]) {viss[d]=1;ff=1;}//如果需要d节点,就代表找到了
for(int i=head[d];i;i=map[i].next)
{
ff=0;
int now=map[i].to;
if(vis[now]) continue;//剪枝
DFS(now);
if(ff) viss[d]=1;//说明这个节点的子树中有需要保留的点,则此节点必须保留
}
if(viss[d]) ff=1;
}
//lca的模板(倍增)
void BFS()
{
d[1]=1;q.push(1);
while(q.size())
{
int x=q.front();q.pop();
for(int i=head[x];i;i=map[i].next)
{
int y=map[i].to;
if(d[y]) continue;
d[y]=d[x]+1;
f[y][0]=x;
for(int j=1;j<=t;++j)
f[y][j]=f[f[y][j-1]][j-1];
q.push(y);
}
}
}
int LCA(int x,int y)
{
if(d[x]>d[y]) swap(x,y);
for(int i=t;i>=0;--i)
if(d[f[y][i]]>=d[x]) y=f[y][i];
if(x==y) return x;
for(int i=t;i>=0;--i)
if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}