LCA(最近公共祖先):给你一棵树,然后给你两个点,让你找到这两个点的最近公共祖先,俩个点的公共祖先有很多,但是其中距离两个点的距离之和最小的那个祖先就是最近公共祖先。
刚学了树上倍增法求LCA
最近做了几个板子题,记录一下这种算法的基本,,,过程吧。
fa数组:倍增关键数组,fa[i][j]表示以i开始找2j步的祖先,例如fa[i][0]就表示i的20的祖先也就是i的父结点。
de数组:当前这个数的在树中的深度。
f数组:记录父节点,有的树没有给出根节点,需要用f数组找根节点,(记录入度数量也可以找到根节点)
head数组:前向星建图头数组
v数组:根结点到当前数的距离(权值)。
首先我们要预处理fa数组,用bfs遍历,更新
f
a
[
i
]
[
j
]
=
f
a
[
f
a
[
i
]
[
j
−
1
]
]
[
j
−
1
]
fa[i][j]=fa[fa[i][j-1]][j-1]
fa[i][j]=fa[fa[i][j−1]][j−1]。同时记录深度,有要求距离的,也在预处理里面更新距离。
然后我们就可以用倍增法查找两个数a,b(de[a]>de[b],默认a是要更深,不然就swap一下)的LCA了。
在求LCA的函数中,我们要先把两个数保持在同一深度,这个过程其实用到的是二进制拆分,把差值一步步缩小,最终让深度相等,然后两个一起往上找祖先,如果两个祖先不一样就同时更新,这样a,b还是在同一深度,最终可以更新到不能一步也不能更新的地步,那么就说明再往前走一步就他俩祖先就一样了,那么这个就是他俩的最近的公共祖先。
例题一:POJ-1330 Nearest Common Ancestors
入门板题
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<math.h>
//#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int N=1e4+50;
int fa[N][20];
int f[N];
int tot;
int head[N];
int de[N];
int L;
int n;
queue<int>q;
struct zxc
{
int to,net;
}e[2*N];
void add(int u,int v)
{
e[tot].to=v;
e[tot].net=head[u];
head[u]=tot++;
}
void init()
{
memset(head,-1,sizeof(head));
memset(f,0,sizeof(f));
memset(fa,0,sizeof(fa));
memset(de,0,sizeof(de));
tot=1;
while(!q.empty())
{
q.pop();
}
}
void bfs(int x)
{
de[x]=1;
q.push(x);
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=head[u];i!=-1;i=e[i].net)
{
int v=e[i].to;
if(de[v])
{
continue;
}
de[v]=de[u]+1;
fa[v][0]=u;
for(int i=1;i<=L;i++)
{
fa[v][i]=fa[fa[v][i-1]][i-1];
}
q.push(v);
}
}
}
int lca(int a,int b)
{
if(de[a]<de[b])
{
swap(a,b);
}
for(int i=L;i>=0;i--)
{
if(de[fa[a][i]]>=de[b])
{
a=fa[a][i];
}
}
if(a==b)
{
return a;
}
for(int i=L;i>=0;i--)
{
if(fa[a][i]!=fa[b][i])
{
a=fa[a][i];
b=fa[b][i];
}
}
return fa[a][0];
}
int main()
{
int t;
int x,y;
scanf("%d",&t);
while(t--)
{
init();
scanf("%d",&n);
L=int(log(n+0.0)/log(2.0))+1;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
f[y]++;
add(x,y);
}
int rot;
for(int i=1;i<=n;i++)
{
if(f[i]==0)
{
rot=i;
break;
}
}
bfs(rot);
scanf("%d%d",&x,&y);
printf("%d\n",lca(x,y));
}
return 0;
}
例题二:POJ-1470 Closest Common Ancestors
//#include<bits/stdc++.h>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<math.h>
#include<map>
using namespace std;
#define LL long long
const double PI=acos(-1.0);
const double eps=1e-8;
const int N=1e5+10;
int fa[1105][12];
//int v[1105];
int de[1105];
int f[1000];
queue<int>q;
struct zxc
{
int to,net;
} e[10005];
int ans[1000];
int head[1005];
int tot;
int n,m,d;
double u;
char s1[5],s2[5];
void add(int u,int v)
{
e[tot]= {v,head[u]};
head[u]=tot++;
// e[tot]={u,w,head[v]};
// head[v]=tot++;
}
void bfs(int x)
{
de[x]=1;
//v[1]=0;
q.push(x);
while(!q.empty())
{
int x=q.front();
q.pop();
for(int i=head[x]; i!=-1; i=e[i].net)
{
int y=e[i].to;
if(de[y])
continue;
de[y]=de[x]+1;
//v[y]=v[x]+e[i].w;
fa[y][0]=x;
for(int i=1; i<=u; i++)
{
fa[y][i]=fa[fa[y][i-1]][i-1];
}
q.push(y);
}
}
}
int lca(int a,int b)
{
if(de[a]<de[b])
{
swap(a,b);
}
for(int i=u; i>=0; i--)
{
if(de[fa[a][i]]>=de[b])
{
a=fa[a][i];
}
}
if(a==b)
{
return a;
}
for(int i=u; i>=0; i--)
{
if(fa[a][i]!=fa[b][i])
{
a=fa[a][i];
b=fa[b][i];
}
}
return fa[a][0];
}
int main()
{
while(~scanf("%d",&n))
{
memset(head,-1,sizeof(head));
tot=1;
memset(fa,0,sizeof(fa));
// memset(v,0,sizeof(v));
memset(de,0,sizeof(de));
memset(ans,0,sizeof(ans));
memset(f,0,sizeof(f));
int x,y,z,num;
u=log(n+0.0)/log(2.0);
for(int i=1; i<=n; i++)
{
scanf("%d:(%d)",&x,&num);
for(int i=1; i<=num; i++)
{
scanf("%d",&y);
f[y]=x;
add(x,y);
}
}
while(f[y])
{
y=f[y];
}
bfs(y);
scanf("%d",&m);
while(m--)
{
scanf("%1s%d%d%1s",s1,&x,&y,s2);
ans[lca(x,y)]++;
}
for(int i=1; i<=n; i++)
{
if(ans[i]!=0)
{
printf("%d:%d\n",i,ans[i]);
}
}
}
return 0;
}
例题三:HDU-4547 CD操作
注意这道题树上从上往下是可以一步到达的。
#include<bits/stdc++.h>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<math.h>
#include<map>
using namespace std;
#define LL long long
const double PI=acos(-1.0);
const double eps=1e-8;
const int N=1e5+10;
int fa[N][20];
int f[N];
int de[N];
int tot;
int n,m;
int u;
char s1[45],s2[45];
map<string,int>q;
queue<int>w;
struct zxc
{
int to,net;
} e[2*N];
int head[N];
void init()
{
for(int i=0;i<=n;i++)
{
head[i]=-1;
f[i]=0;
de[i]=0;
}
memset(fa,0,sizeof(fa));
tot=1;
}
void add(int u,int v)
{
e[tot]= {v,head[u]};
head[u]=tot++;
}
void bfs(int x)
{
de[x]=1;
w.push(x);
while(!w.empty())
{
int z=w.front();
w.pop();
for(int i=head[z]; i!=-1; i=e[i].net)
{
int y=e[i].to;
if(de[y])
{
continue;
}
de[y]=de[z]+1;
fa[y][0]=z;
for(int j=1; j<=u; j++)
{
fa[y][j]=fa[fa[y][j-1]][j-1];
}
w.push(y);
}
}
}
int lca(int x,int y)
{
if(de[x]<de[y])
{
swap(x,y);
}
for(int i=u; i>=0; i--)
{
if(de[fa[x][i]]>=de[y])
{
x=fa[x][i];
}
}
if(x==y)
{
return x;
}
for(int i=u; i>=0; i--)
{
if(fa[x][i]!=fa[y][i])
{
x=fa[x][i];
y=fa[y][i];
}
}
return fa[x][0];
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
q.clear();
int sum=1;
scanf("%d%d",&n,&m);
init();
u=int(log(n+0.0)/log(2.0))+1;
for(int i=1; i<n; i++)
{
scanf("%s%s",s1,s2);
if(!q[s1])
{
q[s1]=sum++;
}
if(!q[s2])
{
q[s2]=sum++;
}
f[q[s1]]=q[s2];
add(q[s2],q[s1]);
}
int num=1;
while(f[num])
{
num=f[num];
}
bfs(num);
while(m--)
{
scanf("%s%s",s1,s2);
int flog=1;
int ans=lca(q[s1],q[s2]);
if(ans==q[s2])
{
flog=0;
}
printf("%d\n",de[q[s1]]-de[ans]+flog);
}
}
return 0;
}