传送门:牛客
题目描述:
小A这次来到一个景区去旅游,景区里面有N个景点,景点之间有N-1条路径。小A从当前的一个景
点移动到下一个景点需要消耗一点的体力值。但是景区里面有两个景点比较特殊,它们之间是可以
直接坐观光缆车通过,不需要消耗体力值。而小A不想走太多的路,所以他希望你能够告诉它,从
当前的位置出发到他想要去的那个地方,他最少要消耗的体力值是多少。
输入:
4
1 2
1 3
2 4
3 4
2
1 3
3 4
输出:
1
0
一道树上距离的题目,需要使用LCA
对于树上 u , v u,v u,v两点之间的距离,我们可以求出两点分别到根节点的距离, d i s t [ u ] , d i s t [ v ] dist[u],dist[v] dist[u],dist[v],然后求出 L C A ( u , v ) LCA(u,v) LCA(u,v)(公共祖先)到根节点的距离 d i s t [ L C A ] dist[LCA] dist[LCA],此时我们的两点间的距离就是 d i s t [ u ] + d i s t [ v ] − 2 ∗ d i s t [ L C A ] dist[u]+dist[v]-2*dist[LCA] dist[u]+dist[v]−2∗dist[LCA],可以画一颗树来理解一下这一部分
对于LCA算法,我们可以使用很多种算法进行解决,博主使用的是树链剖分求LCA的算法,网上有大量讲解这个算法的求解,不乏十分详细的,所以此处就不在赘述了
但是此题并没有解决,对于此题,我们不但需要就出树上两点间的距离,此时我们还有两个点是没有边权的.此时对于这两个特殊点,我们可以使用暴力来判断一下,因为对于这两个点来说,对于当前的<u,v>两个点来说,此时出现的边数也只有 u − a − b − v u-a-b-v u−a−b−v或者 u − b − a − v u-b-a-v u−b−a−v,所以此时我们判断一下这两种另外的情况就行
下面是具体的代码部分:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
ll x=0,w=1;char ch=getchar();
for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int fa[maxn],Size[maxn],max_son[maxn],dep[maxn],top[maxn],dist[maxn];
vector<int>edge[maxn];
void dfs1(int u,int pre_u) {
Size[u]=1;
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==pre_u) continue;
dep[v]=dep[u]+1;dist[v]=dist[u]+1;
dfs1(v,u);
Size[u]+=Size[v];fa[v]=u;
if(Size[v]>Size[max_son[u]]) {
max_son[u]=v;
}
}
}
void dfs2(int u,int t) {
top[u]=t;
if(!max_son[u]) return ;
dfs2(max_son[u],t);
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==fa[u]||v==max_son[u]) continue;
dfs2(v,v);
}
}
int LCA(int u,int v) {
while(top[u]!=top[v]) {
dep[top[u]]>dep[top[v]]?u=fa[top[u]]:v=fa[top[v]];
}
return dep[u]>dep[v]?v:u;
}
int get_dist(int u,int v) {
return dist[u]+dist[v]-2*dist[LCA(u,v)];
}
int n,q;int a,b;
int main() {
n=read();
for(int i=1;i<=n-1;i++) {
int u=read(),v=read();
edge[u].push_back(v);
edge[v].push_back(u);
}
dfs1(1,0);
dfs2(1,1);
a=read();b=read();q=read();
for(int i=1;i<=q;i++) {
int ans=int_INF;
int u=read(),v=read();
int dist1=get_dist(u,a)+get_dist(b,v);
int dist3=get_dist(u,b)+get_dist(a,v);
int dist2=get_dist(u,v);
printf("%d\n",min(min(dist1,dist2),dist3));
}
return 0;
}