轻重链剖分

124 篇文章 0 订阅
#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 500005
using namespace std;

int n,m,s;
int Prev[maxn*2],info[maxn],to[maxn*2],cnt_e;
int fa[maxn],siz[maxn],bel[maxn],tp[maxn],dep[maxn],cnt_l;
int Q[maxn],L,R;
bool vis[maxn];

void insert(int u,int v){
	cnt_e++;
	Prev[cnt_e]=info[u],to[cnt_e]=v,info[u]=cnt_e;
}

void split(int s){
	L=0;
	dep[Q[R=1]=s]=1;
//	fa[s]=-1;
	while(L!=R){
		int now=Q[++L];
		for(int i=info[now];i;i=Prev[i])
			if(!dep[to[i]])
				dep[Q[++R]=to[i]]=dep[fa[to[i]]=now]+1;
	}
	
	for(int i=R;i>0;i--)
	{
		int now=Q[i],son=-1;
		siz[now]=1;
		for(int i=info[now];i;i=Prev[i])
			if(vis[to[i]]){
				siz[now]+=siz[to[i]];
				if(son==-1 || siz[to[i]]>siz[son])
					son=to[i];
			}
		if(son==-1)
		{
			bel[now]=++cnt_l;
			tp[cnt_l]=now;
		}
		else{
			bel[now]=bel[son];
			tp[bel[now]]=now;
		}
		vis[now]=1;
	}
}

int main(){
    scanf("%d%d%d",&n,&m,&s);
    
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        insert(u,v);
        insert(v,u);
    }
    
    split(s);
    
    for(int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        while(bel[u]!=bel[v]){
            if(dep[tp[bel[u]]]<dep[tp[bel[v]]]) v=fa[tp[bel[v]]];
            else u=fa[tp[bel[u]]];
        }
        if(dep[u]<dep[v]) printf("%d\n",u);
        else printf("%d\n",v);
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值