LCA-离线tarjan模板
/* *算法引入: *树上两点的最近公共祖先; *对于有根树的两个结点u,v,最近公共祖先LCA(T,u,v)表示一个结点x,满足x是u,v的祖先且x的深度尽可能大; *对于x来说,从u到v的路径一定经过点x; * *算法思想: *Tarjan_LCA离线算法; *Tarjan算法基于dfs的框架,对于新搜到的一个结点,首先创建由这个结点构成的集合,再对当前结点的每个子树进行搜索; *每搜索完一棵子树,则可确定子树内的LCA询问都已解决,其他的LCA询问的结果必然在这个子树之外; *这时把子树所形成的集合与当前结点的集合合并,并将当前结点设为这个集合的祖先; *之后继续搜索下一棵子树,直到当前结点的所有子树搜完; * *这时把当前结点也设为已被检查过的,同时可以处理有关当前结点的LCA询问; *如果有一个从当前结点到结点v的询问,且v已经被检查过; *则由于进行的是dfs,当前结点与v的最近公共祖先一定还没有被检查; *而这个最近公共祖先的包含v的子树一定已经搜索过了,那么这个最近公共祖先一定是v所在集合的祖先; * *算法步骤: *对于每一个结点: *(1)建立以u为代表元素的集合; *(2)遍历与u相连的结点v,如果没有被访问过,对于v使用Tarjan_LCA算法,结束后将v的集合并入u的集合; *(3)对于与u有关的询问(u,v),如果v被访问过,则结果就是v所在集合的代表元素; * *算法示例: *HDU2586(How far away?) * *题目大意: *求树上任两点间的距离; * *算法思想: *先dfs一遍,求出到根节点的dis; *对于某个询问,求u和v的lca,然后res[i]=d[u]+d[v]-2*d[lca(u,v)]; * **/ struct node{ int to,w,next,lca; }; int fa[maxn]; int head[maxn*2]; int qhead[maxm];//询问 bool vis[maxn]; ll d[maxn]; ll res[maxm]; node edge[N*2]; node qedge[M];//询问边 int n,m; int cnt1,cnt2; int findFa(int x){ return fa[x]==x?x:fa[x]=findFa(fa[x]); } inline void Addedge(int u,int v,int w){ edge[++cnt1].w=w, edge[cnt1].to=v, edge[cnt1].next=head[u], head[u]=cnt1; edge[++cnt1].w=w, edge[cnt1].to=u, edge[cnt1].next=head[v], head[v]=cnt1; } inline void Addqedge(int u,int v){ qedge[++cnt2].to=v, qedge[cnt2].next=qhead[u], qhead[u]=cnt2; //qedge[++cnt2].to=u, qedge[cnt2].next=qhead[v], qhead[v]=cnt2; } void dfs(int u,int fa,ll w){ d[u]=w; for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(v==fa) continue; dfs(v,u,w+edge[i].w); } } void Tarjan_LCA(int u){ //离线LCA算法 fa[u]=u, vis[u]=true; for(int i=head[u];i!=-1;i=edge[i].next){ if(!vis[edge[i].to]){ Tarjan_LCA(edge[i].to); fa[edge[i].to]=u; } } for(int i=qhead[u];i!=-1;i=qedge[i].next){ if(vis[qedge[i].to]){ qedge[i].lca=findFa(qedge[i].to); res[i]=d[u]+d[qedge[i].to]-2*d[qedge[i].lca]; //两者距离 } } } void solve() { rep(i,0,n) fa[i]=i; mem(head,-1); mem(qhead,-1); mem(vis,false); cnt1=cnt2=0; int u,v,w; rep(i,1,n-1){ scanf("%d%d%d",&u,&v,&w); Addedge(u,v,w); } while(m--){ scanf("%d%d",&u,&v); Addqedge(u,v); } dfs(1,-1,0); Tarjan_LCA(1); } int main(){ int T; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); solve(); rep(i,1,m) printf("%lld\n",res[i]); } }