倍增的模板,注释都在上面了

来源:互联网 发布:淘宝退货纸条怎么写 编辑:程序博客网 时间:2024/05/20 02:54
#include<iostream>
#include<cmath>
#define MAXN 500001
using namespace std;
int f[500005][100];
int n,m,s;
int dep[MAXN];
int tot;
int g[MAXN*2];
struct line
{
int to;
int next;
}edge[MAXN*2];
void addedge(int a,int b)
{
tot++;
edge[tot].to=b;
edge[tot].next=g[a];
g[a]=tot;
}
void dfs(int x) //记录深度 
{
for(int i=g[x];i;i=edge[i].next)//邻接表遍历点 
{
if(!dep[edge[i].to])//如果这个深度没被更新,也就是没到过这个点 
{
dep[edge[i].to]=dep[x]+1;
f[edge[i].to][0]=x;//父节点 
dfs(edge[i].to);
}
}
}
void init() //倍增的预处理,f[i][j]就是点i往上走2^j的距离所到的点 
{
dep[s]=1; //根节点的深度为1 
dfs(s); //遍历深度 
for(int j=1;(1<<j)<=n;j++)//这里就是2的j次方要小于n
for(int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
}
int lca(int a,int b)
{
if(dep[a]>dep[b])swap(a,b);//让b是比较深的那个 
int cha=dep[b]-dep[a];//计算深度差 
for(int i=0;cha;cha>>=1,i++)//这里就是把差转化成二进制,比如差值是5(101),那么就是先到f[b][0],然后再到f[b][2],这里差的末位对应的就是i 
if(cha&1)b=f[b][i];
if(a!=b) //开始倍增 
{
for(int i=log(n)/log(2);i>=0;i--)//换底公式,以二为底n的对数 
if(f[a][i]!=f[b][i])
{
a=f[a][i];
b=f[b][i];
}
a=f[a][0];
}
return a;
}
int main()
{
std::ios::sync_with_stdio(false);
cin>>n>>m>>s;
for(int i=1;i<n;i++)
{
int x,y;
cin>>x>>y;
addedge(x,y);
addedge(y,x);
}
init();
for(int i=0;i<m;i++)
{
int x,y;
cin>>x>>y;
cout<<lca(x,y)<<endl;
}
return 0;
}
原创粉丝点击