lca模板,求两点间最短距离


#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e4+7,inf=0x3f3f3f3f,P=80112002;
int n,m,head[N],tot=0,st[N][17],d[N];//(倍增法),链式前向星
ll dis[N];
struct nn{
    int v,nex;
    ll w;
}e[N<<1];//双向边,×2
void adde(int u,int v,ll w){
    e[++tot]={v,head[u],w};head[u]=tot;
}
void dfs(int u,int fa,int dep){//从根往下走,求各个点距根的距离
    d[u]=dep;st[u][0]=fa;//st0是父节点
    for(int i=1;i<=16;i++){
        if((1<<i)<=d[u]){st[u][i]=st[st[u][i-1]][i-1];}
        else{st[u][i]=0;}
    }
    for(int i=head[u];i;i=e[i].nex){
        int v=e[i].v;
        if(v==fa)continue;
        dis[v]=dis[u]+e[i].w;//距所设根的路径唯一,所以长度不用取min
        dfs(v,u,dep+1);
    }
}
int lca(int u,int v){
    if(d[u]<d[v])swap(u,v);
    for(int i=16;i>=0;i--){
        if(d[u]-(1<<i)>=d[v])u=st[u][i];//跳到同一层
    }
    if(u==v)return u;
    for(int i=16;i>=0;i--){
        if(!st[u][i])continue;//也可用d判断
        if(st[u][i]!=st[v][i]){
    //前面已经跳到同层了,所以可以按相同长度跳
    //因为是for且没有break,所以最后肯定i=0,即最后到父结点为lca
            u=st[u][i];
            v=st[v][i];
        }
    }
    return st[u][0];
}
void sol(){
    int x,y;
    ll w;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d%lld",&x,&y,&w);
        adde(x,y,w);adde(y,x,w);
    }
    dfs(1,0,1);//设1为根,dis为到根的距离,dis[1]=1
    while(m--){
        int u,v;
        scanf("%d%d",&u,&v);
        printf("%lld\n",dis[u]+dis[v]-2*dis[lca(u,v)]);
    }
}
int main(){
    int t;
    cin>>t;
    while(t--){
        for(int i=1;i<=n;i++){
            dis[i]=0;d[i]=0;head[i]=0;
            for(int j=0;j<=16;j++)st[i][j]=0;//倍增2^16够了
        }
        tot=0;
        sol();printf("\n");
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值