bzoj 1832: [AHOI2008]聚会 lca

题目大意:

在一棵树上给三个点,求出树上一个点到这三个点的最短距离。

思路:

lca的模板题吧,求出两两之间的lca,最有必然有两对的lca是一样的,那么不同那个点就是最近的那个点了。

程序:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define maxn 500005
using namespace std;
int cnt,head[maxn],f[maxn][17],dep[maxn];
struct data{int to,next;}e[maxn*2];
void add(int x,int y){
    e[++cnt].next=head[x]; head[x]=cnt; e[cnt].to=y;
    e[++cnt].next=head[y]; head[y]=cnt; e[cnt].to=x;
}


void dfs(int x){
    for (int i=1;i<=16;i++){
        if (f[x][i-1]) f[x][i]=f[f[x][i-1]][i-1];
    }
    for (int i=head[x];i;i=e[i].next){
        if (e[i].to==f[x][0]) continue;
        dep[e[i].to]=dep[x]+1;
        f[e[i].to][0]=x;
        dfs(e[i].to);
    }
}

int lca(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=16;i>=0;i--){
        if (dep[f[x][i]]>=dep[y]) x=f[x][i];
    }
    if (x==y) return x;
    for (int i=16;i>=0;i--)
    if (f[x][i]!=f[y][i]){
        x=f[x][i]; y=f[y][i];
    }
    return f[x][0];
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    dep[1]=1;
    dfs(1);
    for (int i=1;i<=m;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        int a,b,c;
        a=lca(x,y); b=lca(y,z); c=lca(x,z);
        if (a==b) printf("%d %d\n",c,-dep[c]+dep[x]+dep[z]-dep[a]*2+dep[y]); else 
        if (b==c) printf("%d %d\n",a,-dep[a]+dep[x]+dep[y]+dep[z]-dep[c]*2); else 
        printf("%d %d\n",b,-dep[b]+dep[x]+dep[y]+dep[z]-dep[c]*2);
    }

} 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值