传送门
题意:求树上的两点的距离,放在一棵树上,而且题目提示只有一条,很明显是这条路线一定会经过他们的最近公共祖先啦,然后就是一个求距离的问题了。如果要采用离线算法的话,就是通过处理所有询问,那么肯定不能单独的求两点之间的距离,我们肯定是要求每一个点到root节点的距离。利用tanjan算法跑一边就可以。
设root为根结点,询问a,b两点距离,a,b的公共祖先为lca点。dis数组为每个点到根节点的距离。
那么ans=dis[a]+dis[b]-2*dis[lca];
具体做法如下
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define me(a,b) memset(a,b,sizeof(a))
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
typedef long long ll;
using namespace std;
const int maxn=4e5+5;
struct node
{
int to,p,val,lca;
}edge[maxn],qedge[maxn];
int vis[maxn],d[maxn],head[maxn],qhead[maxn],f[maxn],s[maxn],e[maxn];
int sign,qsign,n,m;
int find(int x)
{
if(x==f[x]) return x;
else return f[x]=find(f[x]);
}
void add(int u,int v,int val)
{
edge[sign]=node{v,head[u],val};
head[u]=sign++;
}
void qadd(int u,int v)
{
qedge[qsign]=node{v,qhead[u]};
qhead[u]=qsign++;
}
void init()
{
sign=qsign=0;
for(int i=1;i<=n;i++)
{
f[i]=i;
head[i]=qhead[i]=-1;
vis[i]=d[i]=0;
}
}
void dfs(int u)
{
vis[u]=1;
for(int i=head[u];i!=-1;i=edge[i].p)
{
int v=edge[i].to;
if(!vis[v])
{
d[v]=d[u]+edge[i].val;
dfs(v);
f[v]=u;
}
}
for(int i=qhead[u];i!=-1;i=qedge[i].p)
{
int v=qedge[i].to;
if(vis[v])
{
qedge[i].lca=find(v);
if(i%2==0)
qedge[i+1].lca=qedge[i].lca;
else
qedge[i-1].lca=qedge[i].lca;
}
}
}
int main()
{
int t,x,y,z;
scanf("%d",&t);
while(t--)
{
scanf("%d %d",&n,&m);
init();
for(int i=1;i<n;i++)
{
scanf("%d %d %d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
for(int i=1;i<=m;i++)
{
scanf("%d %d",&x,&y);
s[i]=x;
e[i]=y;
qadd(x,y);
qadd(y,x);
}
dfs(1);
for(int i=0;i<m;i++)
{
printf("%d\n",d[s[i+1]]+d[e[i+1]]-2*d[qedge[2*i].lca]);
}
}
return 0;
}