题目描述 Description
小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力
输入描述 Input Description
第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离。
样例输入 Sample Input
3
1 0 1
2 0 1
3
1 0
2 0
1 2
样例输出 Sample Output
1
1
2
数据范围及提示 Data Size & Hint
1<=n<=50000, 1<=m<=75000, 0<=c<=1000
题目说的很详细 求两个点距离父节点的最短距离 也就是求它们的LCA
思想:
两个深度不同的点要找他们的LCA 我们需要让他们先达到同一深度在一起向上跑 知道在第一个相同的点会合
首先 我们需要构建这棵树 用邻接表存图解决 同时记录一下每个点的深度和父节点(每个点从某个点可以访问到 那么这个点就是父节点)子节点的深度 == 父节点深度 + 1
其次 我们就可以进行LCA的过程了 我们默认其中一个点的深度 > 另一个点的深度
我们让深度更大的那个点先向上一直走 更新答案 直到两者深度相同 同时更新当前移动的点为它的父节点
当处于同一深度之后 就可以让两个点一起向上跑 更新答案 同时更新它们所处的位置也就是各自的父节点
注意树根的深度为一
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 50000 + 50;
int n, tot = 0, m;
int first[MAXN], nxt[MAXN << 1], dis[MAXN], deep[MAXN], fa[MAXN];
bool vis[MAXN];
struct edge{
int from, to, cost;
}es[MAXN << 1];
void build(int ff, int tt, int dd)
{
es[++tot] = (edge){ff,tt,dd};
nxt[tot] = first[ff];
first[ff] = tot;
}
void dfs(int x)
{
for(int i = first[x]; i != -1; i = nxt[i])
{
int v = es[i].to;
if(!vis[v])
{
vis[v] = 1;
dis[v] = es[i].cost;
deep[v] = deep[x] + 1;
fa[v] = x;
dfs(v);
}
}
}
int lca(int x, int y)
{
int ans = 0;
if(deep[x] > deep[y])
swap(x,y);
while(deep[y] != deep[x])
{
ans += dis[y];
y = fa[y];
}
while(x != y)
{
ans += dis[x];
ans += dis[y];
x = fa[x];
y = fa[y];
}
return ans;
}
int main()
{
scanf("%d",&n);
memset(first,-1,sizeof(first));
memset(dis,0x3f3f3f3f,sizeof(dis));
for(int i = 1; i < n; i ++)
{
fa[i] = i;
int f, t, d;
scanf("%d%d%d", &f, &t, &d);
build(f,t,d);
build(t,f,d);
}
deep[0] = 1;
vis[0] = 1;
dfs(0);
scanf("%d",&m);
while(m --)
{
scanf("%d%d", &f, &t);
printf("%d\n",lca(f,t));
}
return 0;
}