zoukankan      html  css  js  c++  java
  • lca最近公共祖先(st表/倍增)

    大体思路

    1.求出每个元素在树中的深度

    2.用st表预处理的方法处理出f[i][j],f[i][j]表示元素i上方第2^j行对应的祖先是谁 

    3.将较深的点向上挪,直到两结点的深度相同

    4.深度相同后,祖先可能就在上方,再走几步就到了,于是两个点同时向上移

    具体的方法和代码贴在下面 ↓

    具体

    1.求出每个元素在树中的深度

    //求每个节点在树中的深度
    void dfs(int pos,int pre)//pre是pos的父节点 
    {
        for(int i=0;i<v[pos].size;i++)//枚举pos的子节点 
        {
            register int t=v[pos][i];
            if(t==pre)continue;//防止死循环 
            f[t][0]=pos;dep[t]=dep[pos]+1;
            dfs(t,pos);//再从子节点向后枚举 
        }
    }

    2.用st表预处理的方法处理出f[i][j]

    //求f数组(st表预处理) 
    for(int i=1;(1<<i)<=n;i++)
        for(int j=1;j<=n;j++)
            f[j][i]=f[f[j][i-1]][i-1];
    //f[i][j]表示元素i上方第2^j行对应的祖先是谁 

    3.先比较a,b两点哪个较深,将较深的点赋值给a

    //把a节点变为a,b中较深的一个节点 
    int query(int a,int b)
    {
        if(dep[a]<dep[b])swap(a,b);
    }

    将较深的点向上挪,直到两结点的深度相同

    //使a和b在同一个深度上 
    for(int i=20;i>=0;i--)
        if(dep[f[a][i]]>=dep[b])
            a=f[a][i];
    //倒着循环是因为向上走的步数只会越来越小 

    4.深度相同后,祖先可能就在上方,再走几步就到了,于是两个点同时向上移

    //同一深度后,再向上找公共祖先 
    for(int i=20;i>=0;i--)
        if(f[a][i]!=f[b][i])
        {
            a=f[a][i];
            b=f[b][i];
         } 

     全部代码

    #include <cstdio>
    #include <cstring>
    #include <iostream>
    #include <vector>
    using namespace std;
    vector<int> v[41000];
    vector<int> w[41000];
    int f[41000][21];//f[i][j]表示i点向上2^j层的祖先 
    int g[41000][21];//g[i][j]表示i点到从i向上2^j层的祖先的距离 
    int dep[41000];
    int n,m;
    void dfs(int pos,int pre,int depth)
    {
        dep[pos]=depth;
        for(int i=0;i<v[pos].size();i++)
        {
            int t=v[pos][i];
            if(t==pre) continue;
            f[t][0]=pos;
            g[t][0]=w[pos][i];
            dfs(t,pos,depth+1);
        }
    }
    int query(int a,int b)
    {
        int sum=0;
        if(dep[a]<dep[b]) swap(a,b);//深度较深的点 
        for(int i=20;i>=0;i--)//找到a在深度dep[b]处的祖先 
        {
            if(dep[f[a][i]]>=dep[b])
            {
                sum+=g[a][i];//a到该祖先的距离 
                a=f[a][i];
            }
        }
        if(a==b) return sum;//挪到相同深度后如果在同一点直接return 
        int x;
        for(int i=20;i>=0;i--)//否则a和b一起往上蹦跶 
        {
            if(f[a][i]!=f[b][i])
            {
                sum+=g[a][i];
                a=f[a][i];
                sum+=g[b][i];
                b=f[b][i];
            }
        }
        return sum+g[a][0]+g[b][0];//最后蹦跶到最近公共祖先的下一层,所以要再加上上一层 
    }
    int main()
    {
        int T;
        cin>>T;
        while(T--)
        {
            scanf("%d%d",&n,&m);
            memset(dep,-1,sizeof dep);//多组数据我们初始化 
            memset(f,0,sizeof f);
            memset(g,0,sizeof g);
            for(int i=0;i<n;i++)//md
                v[i].clear(),w[i].clear();
            for(int i=1;i<n;i++)
            {
                int x,y,c;
                cin>>x>>y>>c;
                v[x].push_back(y);
                w[x].push_back(c);
                v[y].push_back(x);
                w[y].push_back(c);
            }
            int xxx=v[1].size();
            dfs(1,0,1);//dfs处理出每个点的深度,以及各种... 
        
            for(int i=1;1<<i<=n;i++)
                for(int j=1;j<=n;j++)
                    f[j][i]=f[f[j][i-1]][i-1],
                    g[j][i]=g[f[j][i-1]][i-1]+g[j][i-1];
            for(int i=1;i<=m;i++)
            {
                int x,y;
                cin>>x>>y;
                if(x==y) cout<<"0"<<endl;
                else cout<<query(x,y)<<endl;
            }
        }
        return 0;
    }
  • 相关阅读:
    iview日期控件获取的数据的转换
    使用vue+iview Form组件 按enter键阻止页面刷新
    vue组件的创建和使用(小功能)
    vue获取本地图片展示到页面上方法
    jQuery ajax
    copy网站小工具
    echarts柱状图颜色设置:echarts柱状图如何设置不同颜色?(代码)
    mysql数据库my.ini配置文件中文详解
    mysql创建流水号
    博客园加入网易云音乐
  • 原文地址:https://www.cnblogs.com/thmyl/p/6067596.html
Copyright © 2011-2022 走看看