AtCoder_ABC368_D - Minimum Steiner Tree题解

题目传送门

看了这个题后，第一反应就是爆搜。

怎么搜呢，很简单，直接从所有 $v[i]$ 的 lca 开始 DFS ，找到了需要的 $v[i]$ 就直接回溯，把一路上的点打上标记，最后统计标记，得到 $ans$ 并输出。

以此题第一个样例为例，需要保留 $1$ ， $3$ ， $5$ ，它们的 lca 为 $1$ ，直接从 $1$ 开始 DFS ，搜到 $5$ 后把 $1$ ， $2$ ， $5$ 标记上，搜到 $3$ 就把 $1$ ， $3$ 标记上， $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];
}