Pasture Walking G 牧场散步 题解

solve

• 貌似是LCA板子题
• 我甚至上来没看懂题在说什么
• 只需要在算dis（路径长）时加上w[i]就行了

Code

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000010;
int n,m,s;
int h[maxn],nxt[maxn],to[maxn],tot,w[maxn];
int Log[maxn];
int dep[maxn];
int dis[maxn];
int fa[maxn][20];
void add(int x,int y,int z)
{
	tot++;
	to[tot]=y;
	w[tot]=z;
	nxt[tot]=h[x];
	h[x]=tot;
}
void llog(){for (int i=2;i<=n;i++) Log[i]=Log[i>>1]+1;}
void dfs(int now,int f)
{
	dep[now]=dep[f]+1;
	fa[now][0]=f;
	for (int i=1;i<=19;i++)
	{
		fa[now][i]=fa[fa[now][i-1]][i-1];
	}
	for (int i=h[now];i;i=nxt[i])
	{
		int y=to[i];
		if (y==f) continue;
		dis[y]=dis[now]+w[i];// 这里qwq
		dfs(y,now);
	}
}
int lca(int x,int y)
{// 纯板子
	if (dep[x]>dep[y]) swap(x,y);
	int d=dep[y]-dep[x];
	for (int i=Log[d];i>=0;i--)
	{
		if (d>>i&1) y=fa[y][i]; 
	}
	if (x == y) return x;
	for (int i=Log[dep[x]];i>=0;i--)
	{
		if (fa[x][i]!=fa[y][i])
		{
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
int main()
{
	cin >> n >> m;
	llog();
	for (int i=1;i<=n-1;i++)
	{
		int x,y,z;
		cin >> x >> y >> z;
		add(x,y,z);
		add(y,x,z);
	}
	dfs(1,1);// 传1就可以
	while (m--)
	{
		int x,y;
		cin >> x >> y;
		int d=dis[x]+dis[y]-dis[lca(x,y)]*2;
		cout << d << endl;
	}
	return 0;
}