bzoj 5165 倍增 and lca

题意:
1.给一个点 加一个儿子编号为n+1
2.询问多个点的LCA

解法:

(1)和两个点LCA类似,找到深度最小的,别的节点都向这这个点条,

在每个节点深度都调整完毕时,如果每个节点都相同那么这个点就是lca
(2)否则,倍增同时向上调整那么最后的所有点的父亲就是LCA

 

#include<bits/stdc++.h>
using namespace std;
#define MAXN 3000010
int MAXLOG;
const int INF = 1e8;
const int inf = 1e8;
int T,tot = 1;
char opt[100];
int q[MAXN],dep[MAXN],anc[MAXN][MAXLOG];
inline void update(int x)
{
    tot+=1;
    dep[tot]=dep[x]+1;
    anc[tot][0]=x;
    for(int i=1;i<MAXLOG;i++){
        anc[tot][i]=anc[anc[tot][i-1]][i-1];
    }
}
inline int query(int k)
{
	int mn=inf;
	for(int i=1;i<=k;i++){
        mn=min(mn,dep[q[i]]);
	}
    for(int i=1;i<=k;i++){
        int t=dep[q[i]]-mn;
        for(int j=0;j<MAXLOG;j++){
            if((t>>j)&1){
                q[i]=anc[q[i]][j];
            }
        }
    }
	///****************************
	bool yes=1;
	for (int i = 1; i <= k; i++) yes &= (q[i] == q[1]);
	if(yes)return q[1];
	for(int i=MAXLOG-1;i>=0;--i){
        bool tem=1;
        for(int j=1;j<=k;j++) tem&=(anc[q[j]][i]==anc[q[1]][i]);
	   if(tem)continue;
        for(int j=1;j<=k;j++) q[j]=anc[q[j]][i];
    }
    return anc[q[1]][0];
}
int main()
{
//freopen("input2.txt","r",        stdin);
	scanf("%d",&T);
	while (T--)
	{int x;
		scanf("%s%d",&opt,&x);
		if (opt[0] == 'A') update(x);
		else
		{
			for (int i = 1; i <= x; i++) scanf("%d",&q[i]);
			printf("%d\n",query(x));
		}
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值