河南萌新联赛2024第（四）场：河南理工大学 I - 马拉松

河南萌新联赛2024第（四）场：河南理工大学 I - 马拉松

题目描述

小明生活在一个名为running的国家，该国家由 n 个城市组成，编号从 1 到 n，还有 n−1 条双向道路连接着这些城市。从任何一个城市都可以到达另一个城市。每条道路都连接着两个城市 a 和 b 。这个国家是一个热爱跑步的国家，因此每天都会在一对城市 (u ，v) 之间举办一次马拉松，从 u 作为起点， v 作为终点，且 u != v 。小明需要用最短的路径到达从 u 到达 v（注意 (u ，v) 和 (v ，u) 是不同的马拉松）。令小明头疼的是该国家有两个城市 x 和 y，当小明经过 x 城市后又跑到 y 城市，那么该城市的警察就会阻止小明继续参加比赛。
小明想知道在举办的所有马拉松路线中他会有多少个马拉松会在他参加时会被警察阻止。

输入描述:

第一行包含三个整数 $n，x，y$ $（2<=n<=3\ast 10^5$ $1<=x，y<=n，x!=y）$–分别是城市数量、x 城市的编号和 y 城镇的编号。
接下来是 $n-1$行，每行包含两个整数 $a$ 和 $b$ $（1<=a，b<=n，a!=b）$
描述了连接两个城镇 a 和 b 的道路。
可以保证的是，从每个城镇出发，我们都可以通过给定的道路到达城市中的其他城镇。也就是说，给定的城镇和道路地图是一棵树。

输出描述:

一个整数，表示小明在参加马拉松时会被警察阻止的数量

输入

5 1 5
1 2
2 3
3 4
4 5

输出

1

两次BFS，标记x可以到达哪些点，y可以到达那些点，他们的各自的子节点相乘即可

#include<bits/stdc++.h>  
using namespace std;  

typedef pair<int,int> PII;
typedef long long LL;

const int N = 300010;
vector<int> g[N];
bool st[N];

int cnt[N];

void bfs(int x,int y,int num)
{
	memset(st,0,sizeof st);
	queue<int> Q;
	Q.push(x);
	
	while(!Q.empty())
	{
		int t=Q.front();
		Q.pop();

        if(st[t]) continue;
		if(t==y) continue;
		
		st[t]=1;
		cnt[t]+=num;
		
		for(auto j:g[t])
			Q.push(j);
	}
} 

signed main() 
{  
    ios::sync_with_stdio(false);
	cin.tie(nullptr);
    
    int n,x,y;cin>>n>>x>>y;
	for(int i=1;i<=n-1;i++)
	{
		int a,b;cin>>a>>b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	
	bfs(x,y,1);
	bfs(y,x,10);
	
	int cnt1=0,cnt2=0;
	for(int i=1;i<=n;i++)
	{
		if(cnt[i]==1) cnt1++;
		else if(cnt[i]==10) cnt2++;
	}	
	
	cout<<(LL)cnt1*cnt2<<"\n";
    
    return 0;
}