bzoj1602[Usaco2008 Oct]牧场行走(lca模版)

Lifel

于 2017-09-05 16:58:01 发布

阅读量358

点赞数

CC 4.0 BY-SA版权

分类专栏： bzoj 树上倍增 LCA 文章标签： bzoj bzoj1602 lca

本文链接：https://blog.csdn.net/Lifel/article/details/77853480

bzoj 同时被 3 个专栏收录

42 篇文章

订阅专栏

权限题

19 篇文章

订阅专栏

LCA

5 篇文章

订阅专栏

本文介绍了一种求解最近公共祖先(LCA)问题的有效算法，并通过一个具体的编程实现案例进行了详细说明。该算法利用预处理技术和深度优先搜索(DFS)来快速定位任意两个节点的最近公共祖先，适用于树形数据结构。此外，还展示了如何通过边集合并的方式建立图的数据结构，以及如何计算节点间的距离。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Lca裸题

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e3+10;
int dep[N],f[N][10],dis[N],
n,q,len=0,last[N];
struct Edge{int to,next,val;Edge(int to=0,int next=0,int val=0):to(to),next(next),val(val){}}e[N<<1];
void add_edge(int u,int v,int w){e[++len]=Edge(v,last[u],w);last[u]=len;}
void dfs(int u,int fa)
{
    for(int i=1;i<=9;i++) {
        if(dep[u]<(1<<i))break;
        f[u][i]=f[f[u][i-1]][i-1];
    }
    for(int i=last[u];i;i=e[i].next){
        int id=e[i].to;
        if(id==fa)continue;
        dep[id]=dep[u]+1;dis[id]=dis[u]+e[i].val;
        f[id][0]=u;
        dfs(id,u);
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=9;i++){
        if(d&(1<<i))x=f[x][i];  
    }
    for(int i=9;i+1;i--) {
        if(f[x][i]!=f[y][i]) 
            x=f[x][i],y=f[y][i];
    }
    if(x==y)return x;
    else return f[x][0];
}
int main()
{
    scanf("%d%d",&n,&q);
    for(int u,v,w,i=1;i<n;i++) {
        scanf("%d%d%d",&u,&v,&w);
        add_edge(u,v,w);add_edge(v,u,w);
    }
    dfs(1,0);
    for(int x,y,i=1;i<=q;i++) {
        scanf("%d%d",&x,&y);
        printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
    }
    return 0;
}