zoukankan      html  css  js  c++  java
  • C++ P3379 【模板】最近公共祖先(LCA)

     题目地址:https://www.luogu.org/problemnew/show/P3379


    主要是用来作为参考代码的。

    #include<cstdio>
    #include<iostream>
    using namespace std;
    int cnt=0,head[1000010],f[500010][21],d[1000010];
    struct Edge{
        int v,nxt;
    }e[1000010];//边 
    void addEdge(int u,int v){
        e[++cnt].v=v;
        e[cnt].nxt=head[u];
        head[u]=cnt;
    }//建图(加边) 
    void dfs(int u,int fa){
        d[u]=d[fa]+1;//初始化点的深度 
        f[u][0]=fa;//结点u的往上第2^0(也就是1)为自己的直接父结点,
        //因此直接设置,并为之后的计算做准备 
        for(int i=1;(1<<i)<=d[u];i++){//循环中i的判断是否可以继续循环状态:2^1 2^2 2^3 
            //也就是从下往上跑深度 
            f[u][i]=f[f[u][i-1]][i-1];
            //乘法原理可得:2^(i-1)+2^(i-1)=2^(i-1)*2=2^i 
        }
        for(int i=head[u];i;i=e[i].nxt){
            if(e[i].v!=fa)dfs(e[i].v,u);
        }//BFS 如果下一个点不是自己的父节点,就继续bfs 
    }
    int lca(int a,int b){
        if(d[a]>d[b])swap(a,b);
        //先保证a的深度小于b的深度,因为可能要将其中一个较深点的位置暂时提到上面 
        //也就是保证b在下面,因为我们判断的是深度
        for(int i=20;i>=0;i--){
            if(d[a]<=d[b]-(1<<i))b=f[b][i];
            //b向上跳(因为我们保证了b在下面),一直到a与b深度相同为止.
            //某不科学的32位int数b:jump jump~
        } 
        if(a==b)return a;
        //如果a和b序号相同,也就是在同一个点上
        //就可以返回a(或b,因为实际是一个点)了 
        for(int i=20;i>=0;i--){
            if(f[a][i]==f[b][i])continue;//同上。如果超了或到了就continue(反正就20次操作) 
            a=f[a][i];b=f[b][i];
        } 
        return f[a][0];
    }
    int main(){
        int n,m,s;
        scanf("%d%d%d",&n,&m,&s);
        for(int i=1;i<n;i++){
        	int ta,tb;
            scanf("%d%d",&ta,&tb);
            addEdge(ta,tb);
            addEdge(tb,ta);
        }//普通的加边建图 
        dfs(s,0);
        for(int i=1;i<=m;i++){
        	int ta,tb;
            scanf("%d%d",&tb,&ta);
            printf("%d
    ",lca(ta,tb));
        }
        return 0;
    } 
  • 相关阅读:
    HPU 1007: 严格递增连续子段(贪心)
    Codeforces Round #224 (Div. 2) A. Ksenia and Pan Scales
    Codeforces Round #224 (Div. 2) A. Ksenia and Pan Scales
    51Nod 1058: N的阶乘的长度(斯特林公式)
    51Nod 1090: 3个数和为0
    CSU 1112: 机器人的指令
    有关刷题时的多组输入问题
    HDU 1060:Leftmost Digit
    《算法导论》— Chapter 6 堆排序
    《算法导论》— Chapter 9 中位数和顺序统计学
  • 原文地址:https://www.cnblogs.com/zbsy-wwx/p/11680701.html
Copyright © 2011-2022 走看看