刚学OI 70分求助
查看原帖
刚学OI 70分求助
137242
Diaоsi楼主2019/10/22 13:36

RT,70分TLE三个点,倍增写的

#include<bits/stdc++.h>
using namespace std;
const int SIZE=1000010;
int f[SIZE][20],d[SIZE],dist[SIZE];
int ver[SIZE<<1],Next[SIZE<<1],edge[SIZE<<1],head[SIZE];
int T,Q,n,m,tot,t;
queue<int> q;
void add(int x,int y){
	ver[++tot]=y,Next[tot]=head[x],head[x]=tot;
}
void bfs(){
	q.push(Q),d[Q]=1;
	while(q.size()){
		int x=q.front();q.pop();
		for(int i=head[x];i;i=Next[i]){
			int y=ver[i];
			if(d[y])continue;
			d[y]=d[x]+1;f[y][0]=x;
			for(int j=1;j<=t;j++)
			    f[y][j]=f[f[y][j-1]][j-1];
			q.push(y);
		}
	}
}
int lca(int x,int y){
	if(d[x]>d[y])swap(x,y);
	for(int i=t;i>=0;i--)
	    if(d[f[y][i]]>=d[x])y=f[y][i];
	if(x==y)return x;
	for(int i=t;i>=0;i--)
	    if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	return f[x][0];
}
int main(){
	cin>>n>>m>>Q;
	t=(int)(log(n)/log(2))+1;
	tot=0;
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	bfs();
	while(m--){
		int x,y;
		cin>>x>>y;
		cout<<lca(x,y)<<endl;
	}
	return 0;
}
2019/10/22 13:36
加载中...