计蒜客信息学赛前 CSP-S2 模拟赛 1 B. 宇宙蓝色闪光

样例输入1

3 1 2
3 1
3 2

样例输出1

1

样例输入2

5 1 3
1 2
2 3
2 4
3 5

样例输出2

2

原题链接

官方题解：

 实在不是很清晰啊

 

【分析】

首先，把A作为根节点遍历，算一下每个点的子树size，然后考虑要使得A被炸到的次数最大，也就是要经过A的次数尽可能多。

对于A的子树son中的每个节点发生爆炸，都有size[son]种情况不会经过A，因此为了经过A的次数最多，我们要选择子树size最小的作为储存室，然后这棵子树的每个点都行

然后B要被经过最少，所以就找size最大的子树，利用哈希计算就好！

 

注意啊，size最大/最小的子树可能不止一个，要全部统计上！！

WA了三次。。。

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
int n,a,b,cnt,head[maxn],fa[maxn],size[maxn];
struct edge
{
	int to,nxt;
}e[maxn<<1];
int ans;
void add(int x,int y)
{
	e[++cnt].nxt=head[x];
	e[cnt].to=y;
	head[x]=cnt;
}
void dfs(int x,int f)
{
	size[x]=1; fa[x]=f;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==f) continue;
		dfs(to,x);
		size[x]+=size[to];
	}
}
int has[maxn];
void dfss(int x)
{ 
	has[x]=1;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==fa[x]) continue;
		dfss(to);
	}
}
void ddfs(int x)
{
	if(has[x]==1) ans++;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==fa[x]) continue;
		ddfs(to);
	}	
}
vector <int> G;
int main()
{
	freopen("halloween.in","r",stdin);
	freopen("halloween.out","w",stdout);
	scanf("%d%d%d",&n,&a,&b);
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y); add(y,x);
	}
	dfs(a,0);
	int sz=0x3f3f3f3f;
	for(int i=head[a];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(size[to]==sz)
			G.push_back(to);
		if(size[to]<sz)
		{
			sz=size[to];
			G.clear();
			G.push_back(to);
		}
		
	}
	has[a]=0;
	has[b]=0;
	for(int k=0;k<G.size();k++)	
		dfss(G[k]);
	memset(fa,0,sizeof(fa));
	memset(size,0,sizeof(size));
	dfs(b,0);
	sz=0;
	G.clear();
	for(int i=head[b];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(size[to]==sz)
			G.push_back(to);
		if(size[to]>sz)
		{
			sz=size[to];
			G.clear();
			G.push_back(to);
		}
	}
	for(int k=0;k<G.size();k++)
		ddfs(G[k]);
	printf("%d\n",ans);
	return 0;
}