zoukankan      html  css  js  c++  java
  • LCA

    #include<iostream>
    #include<cstdlib>
    #include<cstring>
    #include<cstdio>
    #include<algorithm>
    #include<cmath>
    using namespace std;
    const int maxn=50010;
    inline int read()
    {
        int x=0,f=1;
        char ch=getchar();
        while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
        while(isdigit(ch)){x=10*x+ch-'0';ch=getchar();}
        return x*f;
    }
    int first[maxn],to[2*maxn],next[2*maxn],val[2*maxn],cnt;
    int st[maxn][50];
    int vis[maxn];
    int dis[maxn];
    int depth[maxn],f[maxn];
    void add(int a,int b,int c)
    {
        to[++cnt]=b;
        next[cnt]=first[a];
        first[a]=cnt;
        val[cnt]=c;
    }
    void ins(int a,int b,int c){add(a,b,c);add(b,a,c);}
    void dfs(int x)
    {
        st[x][0]=f[x];
        for(int i=1;i<=31;i++)st[x][i]=st[st[x][i-1]][i-1];
        for(int i=first[x];i;i=next[i])
        {
            if(to[i]==f[x])continue;
            f[to[i]]=x;
            depth[to[i]]=depth[x]+1;
            dis[to[i]]=val[i]+dis[x];
            dfs(to[i]);
        }
    }
    pair<int,int> lca(int u,int v)
    {
        int a=u,b=v;
        if(depth[u]<depth[v])swap(u,v),swap(a,b);
        for(int i=31;i>=0;i--)
            if(depth[v]<depth[st[u][i]])u=st[u][i];
        
        if(u==v)return make_pair(v,dis[a]-dis[v]);
        for(int i=31;i>=0;i--)
        {
            if(st[u][i]==st[v][i])break;
            u=st[u][i],v=st[v][i];
        }
        return make_pair(st[u][0],dis[a]+dis[b]-(dis[st[u][0]]<<1));
    }
    int main()
    {
        int n,q;
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            int u=read(),v=read(),w=read();
            ins(u,v,w);
        }
        depth[1]=1;
        vis[1]=1;
        dfs(1);
        q=read();
        pair<int,int> ans;
        for(int i=1;i<=q;i++)
        {
            int u,v;
            u=read(),v=read();
            ans=lca(u,v);
            cout<<ans.first<<" "<<ans.second<<endl;
        }
    }
    倍增
    #include<iostream>
    #include<cstdlib>
    #include<cstring>
    #include<cstdio>
    #include<algorithm>
    #include<cmath>
    using namespace std;
    const int maxn=50010;
    inline int read()
    {
        int x=0,f=1;
        char ch=getchar();
        while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
        while(isdigit(ch)){x=10*x+ch-'0';ch=getchar();}
        return x*f;
    }
    int first[maxn],to[2*maxn],next[2*maxn],val[2*maxn],cnt;
    int qfirst[maxn],qto[2*maxn],qnext[2*maxn],qval[2*maxn],qcnt,qdis[2*maxn],idq[maxn];
    int n,q; 
    int vis[maxn];
    int depth[maxn],fa[maxn];
    int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
    void Unite(int x,int y){if(find(x)!=find(y))fa[find(x)]=find(y);}
    void add(int a,int b)
    {
        to[++cnt]=b;
        next[cnt]=first[a];
        first[a]=cnt;
    }
    void ins(int a,int b){add(a,b);add(b,a);}
    void qadd(int a,int b)
    {
        qto[++qcnt]=b;
        qnext[qcnt]=qfirst[a];
        qfirst[a]=qcnt;
    }
    void qins(int a,int b){qadd(a,b);qadd(b,a);}
    
    void Tarjan_dfs(int x,int f)
    {
        int lca;
        fa[x]=x;
        vis[x]=1;
        for(int i=first[x];i;i=next[i])
        {
            if(to[i]==f)continue;
            vis[to[i]]=1;
            depth[to[i]]=depth[x]+1;
            Tarjan_dfs(to[i],x);
            Unite(to[i],x);
        }
        for(int i=qfirst[x];i;i=qnext[i])
        {
            if(vis[qto[i]])
            {
                lca=find(qto[i]);
                qval[idq[i]]=lca;
                qdis[idq[i]]=depth[qto[i]]+depth[x]-(depth[lca]<<1);
            }
        }
    }
    int main()
    {
        n=read(),q=read();
        int u,v;
        for(int i=1;i<n;i++)
        {
            u=read(),v=read();
            ins(u,v);
        }
        for(int i=1;i<=n;i++)fa[i]=i;
        for(int i=1;i<=q;i++)
        {
            u=read(),v=read();
            qadd(u,v);idq[qcnt]=i;
            qadd(v,u);idq[qcnt]=i;
        }
        depth[1]=1;
        Tarjan_dfs(1,1);
        for(int i=1;i<=q;i++)cout<<qval[i]<<endl;
    }
    Tarjan
  • 相关阅读:
    【JVM】内存与垃圾回收
    【缓存】缓存与分布式锁
    【Redis】入门与基础总结
    【Elasticsearch7.x】Elasticsearch 入门
    iOS开发ReactiveCocoa学习笔记(六)
    iOS开发ReactiveCocoa学习笔记(五)
    iOS开发ReactiveCocoa学习笔记(四)
    iOS开发ReactiveCocoa学习笔记(三)
    iOS开发ReactiveCocoa学习笔记(二)
    iOS开发ReactiveCocoa学习笔记(一)
  • 原文地址:https://www.cnblogs.com/Kong-Ruo/p/7766675.html
Copyright © 2011-2022 走看看