Description
给定一棵 n 个点的树,以及
Input
第一行一个数
接下来 n−1 行,每行三个数 x,y,z ,表示一条从 x 到
第
接下来 m 行,每行两个数
接下来一行一个数 Q ,表示询问次数
接下来 Q 行,每行两个数
Output
Sample Input
4
1 2 5
2 3 2
1 4 3
2
1 2
3 4
1
1 2
Sample Output
5
Solution
线段树维护路径交,树上两条路径 u1↔v1 和 u2↔v2 的交为 lca(u1,u2),lca(u1,v2),lca(v1,u2),lca(v1,v2) 这四个点中深度较深的两点之间的路径,在线倍增后按此合并即可, u↔v 路径长度即为 u 到根的长度+
Code
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
#define maxn 500005
struct edge
{int v,w,next;
}g[2*maxn];
int tot,head[maxn];
void init()
{tot=0;memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{g[tot].v=v,g[tot].w=w,g[tot].next=head[u],head[u]=tot++;
}
int Fa[maxn][21],Deep[maxn];
ll Dis[maxn];
void dfs(int u,int fa)
{Fa[u][0]=fa;for(int i=1;i<=20;i++)Fa[u][i]=Fa[Fa[u][i-1]][i-1];for(int i=head[u];~i;i=g[i].next){int v=g[i].v,w=g[i].w;if(v==fa)continue;Dis[v]=Dis[u]+w,Deep[v]=Deep[u]+1;dfs(v,u);}
}
int Lca(int u,int v)
{if(Deep[u]<Deep[v])swap(u,v);for(int i=20;i>=0;i--)if(Deep[Fa[u][i]]>=Deep[v])u=Fa[u][i];if(u==v)return u;for(int i=20;i>=0;i--)if(Fa[u][i]!=Fa[v][i])u=Fa[u][i],v=Fa[v][i];return Fa[u][0];
}
bool cmp(int a,int b)
{return Deep[a]>Deep[b];
}
P Unite(P x,P y)
{int a[4];a[0]=Lca(x.first,y.first),a[1]=Lca(x.first,y.second),a[2]=Lca(x.second,y.first),a[3]=Lca(x.second,y.second);sort(a,a+4,cmp);return P(a[0],a[1]);
}
#define ls (t<<1)
#define rs (t<<1|1)
P Path[maxn<<2];
void push_up(int t)
{Path[t]=Unite(Path[ls],Path[rs]);
}
void build(int l,int r,int t)
{if(l==r){scanf("%d%d",&Path[t].first,&Path[t].second);return ;}int mid=(l+r)/2;build(l,mid,ls),build(mid+1,r,rs);push_up(t);
}
P query(int L,int R,int l,int r,int t)
{if(L<=l&&r<=R)return Path[t];int mid=(l+r)/2;if(R<=mid)return query(L,R,l,mid,ls);else if(L>mid)return query(L,R,mid+1,r,rs);else return Unite(query(L,R,l,mid,ls),query(L,R,mid+1,r,rs));
}
int main()
{int n,m,q;scanf("%d",&n);init();for(int i=1;i<n;i++){int u,v,w;scanf("%d%d%d",&u,&v,&w);add(u,v,w),add(v,u,w);}dfs(1,1);scanf("%d",&m);build(1,m,1);scanf("%d",&q);while(q--){int l,r;scanf("%d%d",&l,&r);P t=query(l,r,1,m,1);int u=t.first,v=t.second;ll ans=Dis[u]+Dis[v]-2ll*Dis[Lca(u,v)];printf("%lld\n",ans); }return 0;
}