首页 技术 正文
技术 2022年11月11日
0 收藏 593 点赞 3,085 浏览 6510 个字

LCA模板题https://www.luogu.com.cn/problem/P3379题意理解

对于有根树T的两个结点u、v,最近公共祖先LCA(u,v)表示一个结点x,满足x是u、v的祖先且x的深度尽可能小。另一种理解方式是把T理解为一个无向无环图,而LCA(u,v)即u到v的最短路上深度最小的点。
例如,对于下面的树,结点4和结点6的最近公共祖先LCA(T,4,6)为结点2。
【图论算法】LCA最近公共祖先问题
那么这里提供四种思路。

1.暴力至上主义

首先计算出结点u和v的深度d1和d2。如果d1>d2,将u结点向上移动d1-d2步,如果d1<d2,将v结点向上移动d2-d1步,现在u结点和v结点在同一个深度了。下面只需要同时将u,v结点向上移动,直到它们相遇(到达同一个结点)为止,相遇的结点即为u,v结点的最小公共祖先。
但这个算法对于多次询问的题目不能解决。

2.倍增法

思路:倍增法其实是在上一种方法的基础上进行了优化,我们希望向上查找更快,可以事先预处理出p[i,j],表示i往上移动2^j步到达的结点,这样在查找时将大大节约时间。利用P数组可以快速的将结点i向上移动n步,方法是将n表示为2进制数。比如n=6,二进制为110,那么利用P数组先向上移动4步(2^2), 然后再继续移动2步(2^1),即P[ P[i][2] ][1]。
那么首先深搜预处理出所有的p[i][j],并计算每个点的深度d[i]:

void dfs(int u,int fa)//从根结点u开始dfs,u的父亲是fa{
d[u]=d[fa]+; //u的深度为它父亲的深度+1
p[u][]=fa; //u向上走2^0步到达的结点是其父亲
for(int i=;(<<i)<=d[u];i++) p[u][i]=p[p[u][i-]][i-];
//预处理p时,保证能从u走到p[u][i]
for(int i=head[u];i!=-;i=e[i].next)//对于u的每个儿子{
int v=e[i].v;
if(v!=fa)dfs(v,u); //递归处理以v为根结点的子树}
}

然后在主函数中调用dfs(1,0)即可。接下来是查询结点a,b的LCA:

int lca(int a,int b)
{
if(d[a]>d[b]) swap(a,b) ; //保证a点在b点上面
for(int j=;j>=;j--) //将b点向上移动到和a的同一深度
if(d[a]<=d[b]-(<<j)) b=p[b][j] ;
if(a==b) return a ;//如果a和b相遇
for(int j=;j>=;j--)//a,b同时向上移动
{
if(p[a][j]==p[b][j]) continue ;//如果a,b的2^j祖先相同,则不移动
a=p[a][j],b=p[b][j];//否则同时移动到2^j处
}
return p[a][] ;//返回最后a的父亲结点
}

时间复杂度分析
预处理:对每一个结点找到它向上走2^logN步到达的点,所以时间是NlogN
一组询问复杂度:O(logn)。
所以总复杂度为NlogN+QlogN
空间复杂度:O(nlogn)。
完整代码:

#include<stdio.h>
#include<iostream>
#define maxn 500000
using namespace std;int n,m,s,k=;
int dep[maxn+],f[maxn+][];
int head[maxn+];
struct edge{int u,v,next;}e[maxn*+];inline int read()
{
int s=,w=;
char ch=getchar();
while(ch<''||ch>''){if(ch=='-')w=-;ch=getchar();}
while(ch>=''&&ch<='') s=s*+ch-'',ch=getchar();
return s*w;
}void addedge(int u,int v)
{
e[++k].u=u;
e[k].v=v;
e[k].next=head[u];
head[u]=k;
}void deal_first(int u,int father)//递归预处理
{
dep[u]=dep[father]+;
for(int i=;i<=;i++)
f[u][i+]=f[f[u][i]][i];
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].v;
if(v==father) continue;
f[v][]=u;
deal_first(v,u);
}
}int LCA(int x,int y)//查询x,y的LCA
{
if(dep[x]<dep[y]) swap(x,y);//保证x的深度更大
//先暴力,将x和y的深度调至一样
for(int i=;i>=;i--)//x先跳的幅度大一点
{
if(dep[f[x][i]]>=dep[y]) x=f[x][i];
//x向上移动,使得x,y在同一深度
if(x==y) return x;
//如果恰好x的祖先就是y,直接返回x,不过一般不可能
}
for(int i=;i>=;i--)//x,y同时向上移动,找公共祖先
if(f[x][i]!=f[y][i])//x,y的父亲不同才往上跳
x=f[x][i],y=f[y][i];
return f[x][];//最后x和y的父亲一样,返回x的父亲
}int main()
{
// ios::sync_with_stdio(false);
n=read(),m=read(),s=read();
int x,y;
for(int i=;i<=n-;i++)
{
x=read(),y=read();
addedge(x,y);
addedge(y,x);
}
deal_first(s,);//从根节点开始预处理
for(int i=;i<=m;i++)
{
x=read(),y=read();
printf("%d\n",LCA(x,y));
}
return ;
}

3.转为RMQ问题

思路:DFS遍历树T,将遍历到的结点按照顺序记下,得到一个长度为2N – 1的序列,称之为T的欧拉序列F。每个结点都在欧拉序列中出现,我们记录结点u在欧拉序列中第一次出现的位置为pos(u)。如下图:
【图论算法】LCA最近公共祖先问题
根据DFS的性质,对于结点u、v,从pos(u)遍历到pos(v)的过程中经过LCA(u, v)至少一次,且是深度序列B[pos(u)…pos(v)]中值最小的元素在F中对应的节点。那么我们只需要去找在深度序列B中,pos(u)到pos(v)中最小的位置的欧拉序列值。
即LCA( u, v) = RMQ( pos(u), pos(v))。

用occur[]记录T的欧拉序列,depth[]记录深度,first[]记录各结点第一次出现在欧拉序列中的位置,深搜时:

void dfs(int u,int deep)
{
occur[++cnt]=u;//进入该点时进行记录
depth[cnt]=deep;
if(!first[u]) first[u]=cnt;
for(int i=head[u];i!=-;i=e[i].next)
{
dfs(e[i].v,deep+);
occur[++cnt]=u;//访问子树也要标记
depth[cnt]=deep;
}
}

在这之后还需要进行一次预处理,打一次ST表。
时间复杂度分析:dfs时间为2N,RMQ预处理时间为NlogN,询问Q次,所以总时间复杂度为NlogN+Q
完整代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<cstring>
#define maxn 500000
using namespace std;int n,m,s,cnt=,k=,head[maxn+];
struct edge{int u,v,next;}e[maxn*+];
int occur[maxn*+],depth[maxn*+],first[maxn+];
int minnum[maxn*+][],mindep[maxn*+][];
//mindep记录的是区间内深度最小值
//minnum记录的是区间内深度最小值的编号inline int read()//快读
{
int s=,w=;
char ch=getchar();
while(ch<''||ch>''){if(ch=='-')w=-;ch=getchar();}
while(ch>=''&&ch<='') s=s*+ch-'',ch=getchar();
return s*w;
}void addedge(int x,int y)//建边
{
e[++k].u=x;
e[k].v=y;
e[k].next=head[x];
head[x]=k;
}void dfs(int u,int deep)//深搜预处理
{
cnt++;
occur[cnt]=u;
depth[cnt]=deep;
first[u]=cnt;//记录首次出现的位置
for(int i=head[u];i!=-;i=e[i].next)
if(!first[e[i].v])
{
dfs(e[i].v,deep+);//深搜该节点的枝
occur[++cnt]=u;//回溯后,仍要将该节点放入欧拉数列
depth[cnt]=deep;//记录深度
}
}void st()//st表求RMQ
{
for(int i=;i<=cnt;i++)
minnum[i][]=occur[i],mindep[i][]=depth[i];
for(int j=;j<=log(cnt)/log();j++){
for(int i=;i<=cnt-(<<j)+;i++)
{
if(mindep[i][j-]<mindep[i+(<<(j-))][j-])
mindep[i][j]=mindep[i][j-],minnum[i][j]=minnum[i][j-];
else
mindep[i][j]=mindep[i+(<<(j-))][j-],minnum[i][j]=minnum[i+(<<(j-))][j-];
}
}
}int main()
{
memset(head,-,sizeof(head));
n=read(),m=read(),s=read();
int x,y;
for(int i=;i<=n-;i++)
{
x=read(),y=read();
addedge(y,x);
addedge(x,y);
}
dfs(s,);
st();
for(int i=;i<=m;i++)
{
x=read(),y=read();
int op=min(first[x],first[y]),ed=max(first[x],first[y]);
int k=;
k=log(ed-op+)/log();
if(mindep[op][k]<mindep[ed-(<<k)+][k])
printf("%d\n",minnum[op][k]);
else printf("%d\n",minnum[ed-(<<k)+][k]);
}
return ;
}

4.Tarjan

别说了,说多了都是泪。思路:先把所有的询问存下来,然后在DFS的过程中根据点被访问的状态求出每个询问的答案,DFS完后按询问顺序输出。需要用到并查集。
相比前两个算法,这是离线算法。优势在于稳定,且时间复杂度适中。

实现过程
DFS遍历树,利用并查集,当某个节点u及其子树遍历完成后,处理所有与u相关的查询。
1)当遍历到u时,创建u的并查集U={u},集合U的代表为u
2)对u的每一个子树进行DFS遍历,每搜索完一棵子树,将子树标记为checked,把子树所形成的集合与集合U合并,且集合U的代表仍为u
3)按步骤2)继续遍历u的下一棵子树,直到u的所有子树都遍历完,此时将u设为checked
4)处理所有与u相关的查询(u,v)
若v的状态为checked,则LCA(u,v)=present(v)
若v的状态不是checked,跳过这个查询(u,v),当遍历到节点v时,u必然为checked,同样可以求得lca(u,v)

void tarjan(int u)
{
father[u]=u;//以u作为父亲建立一个独立并查集
vis[u]=true;
for(int i=head[u];i!=-;i=e[i].next)
{
int v=e[i].v;
if(!vis[v])
{
tarjan(v);
father[v]=u;
}
}
//深搜寻找叶子节点,再向上回溯
for(int i=;i<q[u].size();i++)
if(vis[q[u][i].first]&&!ans[q[u][i].second])
//如果这个点的并查集已被查询完,并且还没有得出答案
ans[q[u][i].second]=find(q[u][i].first);
//第i号询问就是q[u][i].first的父亲
//同时对并查集的祖先进行刷新
}

时间复杂度分析:Tarjan的时间为N,每个询问O(1),所以总时间为N+Q
完整代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#define maxn 500000
using namespace std;int n,m,s,cnt=,k=,head[maxn+];
struct edge{int u,v,next;}e[maxn*+];
int ans[maxn+],father[maxn+];
bool vis[maxn+];
typedef pair<int,int> pii;
vector<pii> q[maxn+];inline int read()
{
int ans=,flag=;
char ch=getchar();
while(ch<''||ch>'')
{
if(ch=='-') flag=-;
ch=getchar();
}
while(ch>=''&&ch<='')
ans=ans*+ch-'',ch=getchar();
return ans*flag;
}void addedge(int u,int v)
{
e[++k].u=u;
e[k].v=v;
e[k].next=head[u];
head[u]=k;
}int find(int x){return father[x]==x?x:father[x]=find(father[x]);}void tarjan(int u)
{
father[u]=u;//以u作为父亲建立一个独立并查集
vis[u]=true;
for(int i=head[u];i!=-;i=e[i].next)
{
int v=e[i].v;
if(!vis[v])
{
tarjan(v);
father[v]=u;
}
}
//深搜寻找叶子节点,再向上回溯
for(int i=;i<q[u].size();i++)
if(vis[q[u][i].first]&&!ans[q[u][i].second])
//如果这个点的并查集已被查询完,并且还没有得出答案
ans[q[u][i].second]=find(q[u][i].first);
//第i号询问就是q[u][i].first的父亲
//同时对并查集的祖先进行刷新
}int main()
{
memset(head,-,sizeof(head));
n=read(),m=read(),s=read();
int u,v;
for(int i=;i<=n-;i++)
{
u=read(),v=read();
addedge(u,v);
addedge(v,u);
}
for(int i=;i<=m;i++)
{
u=read(),v=read();
q[u].push_back(make_pair(v,i));
q[v].push_back(make_pair(u,i));
}//不定长数组存储询问
tarjan(s);
for(int i=;i<=m;i++)
printf("%d\n",ans[i]);
return ;
}
相关推荐
python开发_常用的python模块及安装方法
adodb:我们领导推荐的数据库连接组件bsddb3:BerkeleyDB的连接组件Cheetah-1.0:我比较喜欢这个版本的cheeta…
日期:2022-11-24 点赞:878 阅读:9,489
Educational Codeforces Round 11 C. Hard Process 二分
C. Hard Process题目连接:http://www.codeforces.com/contest/660/problem/CDes…
日期:2022-11-24 点赞:807 阅读:5,904
下载Ubuntn 17.04 内核源代码
zengkefu@server1:/usr/src$ uname -aLinux server1 4.10.0-19-generic #21…
日期:2022-11-24 点赞:569 阅读:6,737
可用Active Desktop Calendar V7.86 注册码序列号
可用Active Desktop Calendar V7.86 注册码序列号Name: www.greendown.cn Code: &nb…
日期:2022-11-24 点赞:733 阅读:6,490
Android调用系统相机、自定义相机、处理大图片
Android调用系统相机和自定义相机实例本博文主要是介绍了android上使用相机进行拍照并显示的两种方式,并且由于涉及到要把拍到的照片显…
日期:2022-11-24 点赞:512 阅读:8,128
Struts的使用
一、Struts2的获取  Struts的官方网站为:http://struts.apache.org/  下载完Struts2的jar包,…
日期:2022-11-24 点赞:671 阅读:5,291