树上倍增 0pts TLE 求调!
查看原帖
树上倍增 0pts TLE 求调!
373198
WhileTrueRP楼主2023/9/14 14:22
#include<iostream>
#include<vector>
#include<utility>
#include<cstdio>
using namespace std;
const int N = 1e6+5;
vector<int> a[N];
int n,m,x,y;
int d[N],s[N],anc[N][25];
void add(int x,int y){
	a[x].push_back(y);
	a[y].push_back(x);
}
//void dfs(int now,int fa){
//	d[now] = d[fa] + 1;
//	s[now] = s[fa] + a[now].size();
//	for(int i=0;i<a[now].size();i++){
//		int son = a[now][i];
//		if(son != fa){
//			anc[now][0] = fa;
//			dfs(son,now);
//		}
//	}
//}
void dfs(int u,int fa){
	for(int i=0;i<a[u].size();i++){
		int v = a[u][i];
		if(v == fa){
			continue;
		}
		d[v] = d[u] + 1;
		s[v] = s[u] + a[v].size()/2;
		anc[v][0] = u;
		dfs(v,u);
	} 
}
void init(){
	for(int j=1;j<=18;j++){
		for(int i=1;i<=n;i++){
			anc[i][j] = anc[anc[i][j-1]][j-1];
		}
	}
}
int lca(int u,int v){
	if(d[u] < d[v]){ 
		swap(u,v);
	}
	for(int i=18;i>=0;i--){
		if(d[anc[u][i]] >= d[v]){
			u = anc[u][i];
		}
	}
	if(u == v){
		return u;
	}
	for(int i=18;i>=0;i--){
		if(anc[u][i] != anc[v][i]){
			u = anc[u][i];
			v = anc[v][i];
		}
	}
	return anc[u][0];
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n-1;i++){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);	
	}
//	return 1;
	dfs(1,0);
	init();
	for(int i=1;i<=m;i++){
		scanf("%d%d",&x,&y);
		if(x == y){
			printf("%d",a[x].size()/2);
			continue;
		}
		int l = lca(x,y);
		printf("%d\n",s[x]+s[y]-s[l]*2+a[l].size()/2);
	}
	return 0;
}
2023/9/14 14:22
加载中...