关于st表构建时数组第二维的枚举范围
查看原帖
关于st表构建时数组第二维的枚举范围
213141
谁伴我流浪楼主2024/11/4 16:40

为什么在第一遍dfs构建st表时p的范围用(1<<p)<=deep[x]就可以AC但是改成p<=deep[x]就会WA

#include<bits/stdc++.h>
#define N 2500000
#define ll long long
using namespace std;
int n,m,tot;
int head[N],to[N],nxt[N];
int deep[N],st[N>>1][20];
int vis[N>>1];
int power;
void add(int x,int y){
	to[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
}
void dfs(int x,int fa){
	deep[x]=deep[fa]+1;
	st[x][0]=fa;
	vis[x]=1;
	for(int p=1;(1<<p)<=deep[x];p++)
		st[x][p]=st[st[x][p-1]][p-1];
	for(int i=head[x];i;i=nxt[i]){
		int v=to[i];
		if(!vis[v]){
			dfs(v,x);
		}
	}
}
int log_2(int x){
	int ans=0;
	while((1<<ans)<=(x>>1))
		ans++;
	return ans;
}
int lca(int a,int b){
	if(deep[a]<deep[b]) swap(a,b);
	for(int p=power;p>=0;p--){
		if(deep[st[a][p]]>=deep[b]) a=st[a][p];
	}
	if(a==b) return a;
	for(int p=power;p>=0;p--){
		if(st[a][p]!=st[b][p]){
			a=st[a][p];
			b=st[b][p];
		}
	}
	return st[a][0];
}
int main() {
	cin>>n>>m;
	power=log_2(n);
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	for(int i=1;i<=m;i++){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		int f1=lca(a,b),f2=lca(a,c),f3=lca(b,c);
		int high= f1!=f2?(deep[f1]<deep[f2]?f1:f2):f1;
		int low= f1!=f2?(deep[f1]>deep[f2]?f1:f2):f3;
		printf("%d %lld\n",low,(ll)(deep[a]+deep[b]+deep[c]-deep[high]*2-deep[low]));
	}
	return 0;
}
2024/11/4 16:40
加载中...