Distance Queries（距离查询）

Distance Queries（距离查询）

原题目网址（可提交检查）

题目描述

农民约翰的奶牛拒绝在跑他的马拉松，因为他选择的路径太长不适合奶牛们悠闲的生活方式。因此，他希望找到一个更合理的路径长度。农民约翰要查询一系列点对间的路径长度，请尽快回答农民约翰的询问！

输入输出格式

输入格式：

第一行包括两个整数N，M，N (2 <= N <= 40，000)表示农场数量，每个农场标记为1-N，M (1 <= M < 40，000)表示路径数量。
　　第二至M+1行，每行包括三个整数F1，F2，L和一个字符D，表示农场F1和F2之间的路径长度为L，D为’N’， ‘E’， ‘S’， 'W’中的一个，表示从F1到F2的方向。
　　第M+2行包括一个整数K，1 <= K <= 10，000
　　第M+3至M+K+2行，每行包括两个整数，表示要询问的两个农场。

输出格式：

共K行，每行输出每个询问的路径长度。

输入输出样例

输入样例：

7 6
1 6 13 E
6 3 9 E
3 5 7 S
4 1 3 N
2 4 20 W
4 7 2 S
3
1 6
1 4
2 6

输出样例：

13
3
36

Code 参考代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=44000,M=110000;
long long n,m,l[N],ans[N][50],d[N],dis[N],jl[N],ne[M],to[M],w[M],cnt=0;
void add(int x,int y,int z)
{
	cnt++;
	to[cnt]=y;
	ne[cnt]=jl[x];
	jl[x]=cnt;
	w[cnt]=z;
}
void build(int x)
{
	dis[x]=dis[ans[x][0]]+1;
	for(int i=jl[x];i;i=ne[i])
		if(to[i]!=ans[x][0])
		{
			ans[to[i]][0]=x;
			d[to[i]]=d[x]+w[i];
			build(to[i]);
		}
}
int lca(int x,int y)
{
	if(dis[x]<dis[y])
		swap(x,y);
	int dd=dis[x]-dis[y];
	for(int i=l[dd];i>=0;i--)
		if(dd&(1<<i))
			x=ans[x][i];
	for(int i=l[n];i>=0;i--)
		if(ans[x][i]!=ans[y][i])
		{
			x=ans[x][i];
			y=ans[y][i];
		}
	if(x!=y)
		return ans[x][0];
	return x;
}
int main()
{
	char c[10];
	long long x,y,z,k;
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=m;i++)
	{
		//cout<<i<<":";  检验
		scanf("%lld%lld%lld%s",&x,&y,&z,c);
		//cout<<x<<" "<<y<<" "<<z<<" "<<c<<endl;  检验
		add(x,y,z);
		add(y,x,z);
	}
	//cout<<2222222;  检验
	build(1);
	l[0]=-1;
	for(int i=1;i<=n;i++)
		l[i]=l[i>>1]+1;
	for(int i=1;i<=l[n];i++)
		for(int j=1;j<=n;j++)
			ans[j][i]=ans[ans[j][i-1]][i-1];
	scanf("%lld",&k);
	for(int i=1;i<=k;i++)
	{
		scanf("%lld%lld",&x,&y);
		printf("%lld\n",d[x]+d[y]-d[lca(x,y)]*2);
	}
	return 0;
}

原题目网址（可提交检查）

结束语

谢谢浏览，点赞收藏后再走，留下您的痕迹！😀