倍增法求LCA

一般用于求解树上两结点距离。LCA是两节点的最近公共祖先。

倍增法求解LCA实际上是将上升的步数用二进制来表示,经过向上跳两结点深度的比较来确定是否可以上升,最终找到LCA。


先建树维护几个数组

father[][];   记录祖先结点编号 
deep[];       记录结点深度
dis[][]记录结点和祖先结点的距离 

mxdeep 代表最大层数   mxdeep=log(n)/log(2);

代码:

void dfs(int root)
{
	for(int i=1;i<=mxdeep;i++)
	{
		father[root][i]=father[father[root][i-1]][i-1];
		dis[root][i]=dis[father[root][i-1]][i-1]+dis[root][i-1];
		if(!father[root][i])   //优化,当为0的时候说明已经更新到了根结点位置 
		break;
	}
	for(int i=0;i<V[root].size();i++)
	{
		tree &e=G[V[root][i]];
		if(e.to!=father[root][0]) //说明e.to是root的儿子
		{
			deep[e.to]=deep[root]+1;
			father[e.to][0]=root;  //建立父子关系
			dis[e.to][0]=e.w;      //更新到父亲的距离
			dfs(e.to); 
		} 
	} 
}

也可以把维护father过程单独拿出来写法类似RMQ

void beizeng()  
{
	for(int i=1;i<=mxdeep;i++)
	{
		for(int j=1;j<=n;j++)
		{
			father[j][i]=father[father[j][i-1]][i-1];
			dis[j][i]=dis[father[j][i-1]][i-1]+dis[j][i-1];
		}
	}
}


这里说一下为什么 father[root][i]=father[father[root][i-1]][i-1]。

因为father[root][i]代表 结点root的第 2^i 倍祖先的结点编号,所以father[root][0] 代表着root的父亲。这里的倍并不代表倍数而是距离,所以 2^(i-1)+2^(i-1)=2^i。


接着让深度大的先上移,但不能超过深度小的,之后如果没有移动到一起,再两个一起上移。

最后可能出现   root

                      u       v

那么答案就是u或者v的父亲。

LCA代码:

int lca(int u,int v)
{
	int ans=0;
	if(deep[u]>deep[v])  //让u结点不在v结点下方,方便操作 
	swap(u,v);
	for(int i=mxdeep;i>=0;i--)  //将 v结点上移,但不能高于u结点 
	{
		if(deep[u]<deep[v]&&deep[u]<=deep[father[v][i]])
		{
		//	printf("! %d %d %d %d\n",u,v,i,father[v][i]);
			ans+=dis[v][i];
			v=father[v][i];
		}
	}
	if(v==u)
	return ans;
	for(int i=mxdeep;i>=0;i--)
	{
		if(father[v][i]!=father[u][i])  //一起移动到共同父亲处 
		{
			ans+=dis[v][i]+dis[u][i];
			v=father[v][i];
			u=father[u][i];
		}
	}
	if(v!=u) ans+=dis[v][0]+dis[u][0];  // 如果没有移动到同一处,可能是   1->u   1->v ,那么向上移动一个位置就可以了 
	return ans;
}



                
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值