zoukankan      html  css  js  c++  java
  • LCA最近公共祖先

    1.ST表+时间戳


    DFS记录这棵树每个点出现的时间1 2 3 2 4 2 5 2 1 6 7 8 7 6 1。
    然后用ST表记录每个区间的min值,然后进行q次查询,时间复杂度O(Nlog2N+Qlog2N);

    #include<iostream>
    #include<stdio.h>
    #include<stdlib.h>
    #include <iomanip>
    #include<cmath>
    #include<float.h> 
    #include<string.h>
    #include<algorithm>
    #define sf scanf
    #define scf(x) scanf("%d",&x)
    #define scff(x,y) scanf("%d%d",&x,&y)
    #define pf printf
    #define prf(x) printf("%d
    ",x)
    #define mm(x,b) memset((x),(b),sizeof(x))
    #include<vector>
    #include<queue>
    #include<map>
    #define rep(i,a,n) for (int i=a;i<n;i++)
    #define per(i,a,n) for (int i=a;i>=n;i--)
    typedef long long ll;
    const ll mod=1e9+100;
    const double eps=1e-8;
    using namespace std;
    const double pi=acos(-1.0);
    const int inf=0xfffffff;
    const int N = 1010;
    int rmq[2*N];//rmq数组,就是欧拉序列对应的深度序列
    struct ST
    {
        int mm[2*N];
        int dp[2*N][20];//最小值对应的下标
        void init(int n)
        {
            mm[0] = -1;
            for(int i = 1;i <= n;i++)
            {
                mm[i] = ((i&(i-1)) == 0)?mm[i-1]+1:mm[i-1];
                dp[i][0] = i;
            }
            for(int j = 1; j <= mm[n];j++)
                for(int i = 1; i + (1<<j) - 1 <= n; i++)
                    dp[i][j] = rmq[dp[i][j-1]] < rmq[dp[i+(1<<(j-1))][j-1]]?dp[i][j-1]:dp[i+(1<<(j-1))][j-1];
        }
        int query(int a,int b)//查询[a,b]之间最小值的下标
        {
            if(a > b) swap(a,b);
            int k = mm[b-a+1];
            return rmq[dp[a][k]] <= rmq[dp[b-(1<<k)+1][k]]?dp[a][k]:dp[b-(1<<k)+1][k];
        }
    };
    //边的结构体定义
    struct Edge
    {
        int to,next;
    };
    Edge edge[N*2];
    int tot,head[N];
    
    int F[N*2];//欧拉序列,就是dfs遍历的顺序,长度为2*n-1,下标从1开始
    int P[N];//P[i]表示点i在F中第一次出现的位置
    int cnt;
    
    ST st;
    void init()
    {
        tot = 0;
        memset(head,-1,sizeof(head));
    }
    void addedge(int u,int v)//加边,无向边需要加两次
    {
        edge[tot].to = v;
        edge[tot].next = head[u];
        head[u] = tot++;
    }
    void dfs(int u,int pre,int dep)
    {
        F[++cnt] = u;
        rmq[cnt] = dep;
        P[u] = cnt;
        for(int i = head[u];i != -1;i = edge[i].next)
        {
            int v = edge[i].to;
            if(v == pre)continue;
            dfs(v,u,dep+1);
            F[++cnt] = u;
            rmq[cnt] = dep;
        }
    }
    void LCA_init(int root,int node_num)//查询LCA前的初始化
    {
        cnt = 0;
        dfs(root,root,0);
        st.init(2*node_num-1);
    }
    int query_lca(int u,int v)//查询u,v的lca编号
    {
        return F[st.query(P[u],P[v])];
    }
    bool root[N];
    int main()
    {
    	int n,m,num,v,u;
    	while(~scff(n,m))//n个点,m个查询点 
    	{
    		init();
    		mm(root,true);
    		rep(i,1,n)
    		{
    			sf("%d %d",&u,&v);
    				addedge(u,v);
    				addedge(v,u);
    				root[v]=false;
    		}
    		int temp;
    		rep(i,1,n+1)
    		{
    			if(root[i])
    			{
    				temp=i;break;
    			}
    		}
    		LCA_init(temp,n);
    		while(m--)
    		{
                scanf("%d%d",&u,&v);
                prf(query_lca(u,v));
    		}
    	
    	}
    	return 0;
    }
    

    2.targin

    targin的多个查询的话还不会,似乎也可以的,后面学学要怎么做

    const int N=1e4+3;
    int node[N]; 
    bool root[N];
    int cnt;
    int visit[N];
    int fa[N];
    int le,ri;
    struct Edge
    {
    	int v,next;
    }edge[N];
    void add_edge(int u,int v)
    {
    	edge[cnt].next=node[u];
    	edge[cnt].v=v;
    	node[u]=cnt++;
    }
    int find(int x)
    {
    	if(fa[x]==x) return x;
    	return fa[x]=find(fa[x]);
    }
    void Union(int x,int y)
    {
    	x=find(x);
    	y=find(y);
    	if(x!=y)
    	fa[y]=x;
    }
    void tarjin(int x)
    {
    	for(int i=node[x];i!=-1;i=edge[i].next)
    	{
    		tarjin(edge[i].v);
    		Union(x,edge[i].v); 
    	}
    	visit[x]=1;
    	if(x==le&&visit[ri])
    	{
    		prf(find(ri));
    		return ;
    	}
    	if(x==ri&&visit[le])
    	{
    		prf(find(le));
    		return ;
    	}
    }
    int main()
    {
    	int re,x,y;
    	scf(re);
    	while(re--)
    	{
    		mm(fa,-1);
    		mm(root,true);
    		mm(node,-1);
    		mm(visit,0);
    		int n;
    		cnt=0;
    		scf(n);
    		rep(i,1,n)
    		{
    			scff(x,y);
    			root[y]=false;
    			fa[i]=i;
    			add_edge(x,y);
    		}
    		fa[n]=n;
    		scff(le,ri);
    		rep(i,1,n+1)
    		if(root[i])
    		{
    			tarjin(i);
    			break;
    		}
    	}
    	return 0;
    }
    
  • 相关阅读:
    caffe:mac10.12安装caffe的步骤
    查找两个链表的共同子链表
    golang:1.并发编程之互斥锁、读写锁详解
    git问题汇总
    有用的技术工具
    maven 安装本地jar包到本地maven仓库
    win7下Hadoop学习 之 Cygwin下载、安装、配置
    简单目录备份脚本
    2021年01月28日微博热搜汇总
    2021年01月26日微博热搜汇总
  • 原文地址:https://www.cnblogs.com/wzl19981116/p/9529435.html
Copyright © 2011-2022 走看看