今天,杨子来曰(yue)一曰(yue)算法——倍增求LCA
LCA是神马呢?
举个例子,我姓杨,你也姓杨,所以我们早在5325年前肯定有一个公共祖先,BUT这个公共祖先的儿子也有可能是我们公共的祖先,对吧?So,肯定在我们的所有公共祖先中,有一个是离我们最近的(也就是说,他的两个儿子分别是我和你的祖先),那么这个祖先就称为,我和你的最近公共祖先,即LCA.
现在,我们把他放在一棵树上就很好理解了,Look at the图:
如图:结点6和结点8的LCA就是2
再看:
如图:结点7和结点9的LCA是结点1
但如果是这样呢?
没错,结点1和结点9的LCA就是结点1,So,杨子曰:如果两个结点是爹和娃的关系,那么它们的LCA就是爹
好了,BB了半天,看出来点什么呢?或者问,我们为什么要研究LCA呢?
来看一看两点间的路径,你可以得到一个结论,杨子曰:树上两点间的路径必定经过他们的LCA
So,杨子再曰:有关两点间路径的问题,立刻想到LCA(也有可能是树链剖分)
呱呱了半天,LCA到底怎么求呢?——三个算法,今天就曰一个——倍增
(可以先看一下:ST表——杨子曰数据结构,有助于理解)
首先,我们需要一个数组f[i][j],表示结点i的第2^j个父亲是谁,那f怎样更新呢?
首先f数组全附成根节点(原因后面自然就知道了)f[i][0]可以用一次dfs全部求出,然后会发现f[i][j]就等于f[f[i][j-1]][j-1],也就是结点i的第2^ j的父亲就是i的第2^ (j-1)父亲的第2^(j-1)父亲
注意i,j循环的顺序:
for (int j=1;j<=18;j<=++){
for (int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
}
}
OK,这个求完了再怎么搞呢?
下面,我们假设要求x,y两个点的LCA
总结一波:先dfs求f[i][0]和时间戳,递推求出f[i][j],对于每个询问单独判断,时间复杂度:O(n log n)
如果你还知道神马是树链剖分的话(不知道,戳我),你就可以用树链剖分求LCA(←它比倍增更快更优)
OK,完事
c++代码:
#include
using namespace std;
int f[30005][20],tin[30005],tout[30005];
vector<int> a[30005];
int cnt=0;
int inline read(){
int x=0,w=0;char ch=0;
while (ch<'0' || ch>'9') w|=ch=='-',ch=getchar();
while (ch>='0' && ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return w? -x:x;
}
void dfs(int k,int fa){
f[k][0]=fa;
tin[k]=++cnt;
for (int i=0;i<a[k].size();i++){
int v=a[k][i];
if (v==fa) continue;
dfs(v,k);
}
tout[k]=cnt;
}
int anc(int x,int y){
return tin[x]<=tin[y] && tout[y]<=tout[x];
}
int lca(int x,int y){
if (anc(x,y)) return x;
if (anc(y,x)) return y;
for (int i=16;i>=0;i--){
if (!anc(f[x][i],y)) x=f[x][i];
}
return f[x][0];
}
int main(){
int n=read();
for(int i=1;i<n;i++){
int x=read();
int y=read();
a[x].push_back(y);
a[y].push_back(x);
}
dfs(1,1);
for (int j=1;j<=16;j++){
for (int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
}
}
int m;
m=read();
while(m--){
int x,y;
x=read();
y=read();
cout<<lca(x,y)<<"\n";
}
return 0;
}
推荐一道题:【codevs3305】水果姐逛水果街Ⅱ
于TJQ高层小区
未经作者允许,严禁转载:https://blog.csdn.net/henryyang2018/article/details/79692080