ZOJ 3195 LCA转RMQ

点击打开链接

题意：输入n，接下来n-1行，每行a，b，c代表a与b有一条权值为c的边，双向边，m次询问，问最短的距离使a，b，c可以联通

思路：LCA的模版题，没什么好说的，看理论的话网上好多本弱就不说了，代码有注释，有助于理解

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=50010;
struct edge{
    int to,cost;
    edge(int a,int b){to=a;cost=b;}
};
vector<edge>G[maxn];
bool vis[maxn];
int L[maxn*2],E[maxn*2],H[maxn],dis[maxn],dp[2*maxn][20];
//H为每个元素第一次出现的位置，E为dfs遍历的序号，L为深度，dis为每个点到根的距离
//dp[i][j]代表的是i到j深度最小的位置
int k,n;
void dfs(int t,int deep){
    k++;E[k]=t;L[k]=deep;H[t]=k;//遍历时叶子在E中只会出现一次，而非叶子节点会出现多次
    for(unsigned int i=0;i<G[t].size();i++){
        edge tt=G[t][i];
        if(!vis[tt.to]){//出现过就不再继续，保证dis的大小和H第一次出现的位置
            vis[tt.to]=1;
            dis[tt.to]=dis[t]+tt.cost;
            dfs(tt.to,deep+1);
            k++;E[k]=t;L[k]=deep;
        }
    }
}
void RMQ_init(){//RMQ的ST算法
    for(int i=1;i<=2*n-1;i++) dp[i][0]=i;
    for(int i=1;(1<<i)<=2*n-1;i++){
        for(int j=1;j+(1<<i)-1<=2*n-1;j++){
            if(L[dp[j][i-1]]<L[dp[j+(1<<(i-1))][i-1]]) dp[j][i]=dp[j][i-1];
            else dp[j][i]=dp[j+(1<<(i-1))][i-1];
        }
    }
}
int RMQ(int le,int ri){
    le=H[le];ri=H[ri];//找到第一次出现的位置
    if(le>ri) swap(le,ri);//le大于ri说明ri在le之前出现，交换
    int kk=0;
    while((1<<(kk+1))<=ri-le+1) kk++;
    if(L[dp[le][kk]]<L[dp[ri-(1<<kk)+1][kk]]) return E[dp[le][kk]];
    else return E[dp[ri-(1<<kk)+1][kk]];//返回的是dp位置的值
}
int main(){
    int q,a,b,c;
    int flag=0;
    while(scanf("%d",&n)!=-1){
        for(int i=0;i<maxn;i++) G[i].clear();
        if(flag==0) flag=1;
        else printf("\n");
        memset(dis,0,sizeof(dis));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n-1;i++){
            scanf("%d%d%d",&a,&b,&c);
            G[a].push_back(edge(b,c));
            G[b].push_back(edge(a,c));
        }
        k=0;vis[0]=1;
        dfs(0,1);RMQ_init();
        scanf("%d",&q);
        while(q--){
            scanf("%d%d%d",&a,&b,&c);
            int ans=dis[a]+dis[b]+dis[c]-(dis[RMQ(a,b)]+dis[RMQ(a,c)]+dis[RMQ(b,c)]);
            //这个画图看一下就能看出来了
            printf("%d\n",ans);
        }
    }
    return 0;
}