AtCoderD - Collision bfs+预处理

Posted 2021-11-04 karshey

D
参考

权值都是1，典型bfs；
每次都bfs会T，可以bfs点1到每个点的距离（预处理）。
若想知道3到5是在Road还是Town，只需知道1-3和3-5的距离之和。
偶数是Town，奇数是Road；

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define debug(x) cout<<#x<<" "<<x<<endl;
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
//边权相等 求最短路的长度，长度为偶数就是Town======================
const int N=1e5+10;
int n,qu;
int s,e;
vector<int>a[N];
queue<int>q;
int dis[N];
void bfs()
{
	mem(dis,-1);
	dis[s]=0;
	q.push(1);
	while(q.size())
	{
		int t=q.front();q.pop();
		for(int i=0;i<a[t].size();i++)
		{
			int temp=a[t][i];						
			if(dis[temp]==-1) //没有到过					
			{
				dis[temp]=dis[t]+1;
				q.push(temp);
			}			
		}						
	}
}
int main()
{
	cin>>n>>qu;
	fir(i,1,n-1)
	{
		int aa,bb;scanf("%d%d",&aa,&bb);
		a[aa].pb(bb);
		a[bb].pb(aa);
	}
	bfs();
	while(qu--)
	{
		scanf("%d%d",&s,&e);
		int ans=(dis[s]+dis[e]);
		if(ans&1) cout<<"Road"<<endl;
		else cout<<"Town"<<endl;
	}
	return 0; 
}