非常坏LCA,使我的倍增旋转
查看原帖
非常坏LCA,使我的倍增旋转
846661
ARIS1_0楼主2024/1/8 13:57

WA on #5,6

#include<bits/stdc++.h>
#define ll long long
using namespace std;
mt19937 myrand(time(0));
inline ll read(){
	ll x=0,w=1;
	char ch=0;
	while(ch<'0'||ch>'9'){
		if(ch=='-')w=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		x=(x<<3)+(x<<1)+(ch^48);
		ch=getchar();
	}
	return x*w;
}
void write(ll x){
	if(x<0){
		putchar('-');
		x=-x;
	}
	static int sta[35];
	int top=0;
	do{
		sta[top++]=x%10,x/=10;
	}while(x);
	while(top)putchar(sta[--top]+'0');
}
vector<int>node[105];
int n,dep[105],x,y,dp[105][20],lg[205];
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	dp[u][0]=fa;
	for(int i=1;i<=lg[dep[u]];i++){
		dp[u][i]=dp[dp[u][i-1]][i-1];
	}
	for(int i=0;i<node[u].size();i++){
		if(node[u][i]==fa)continue;
		dfs(node[u][i],u);
	}
}
int LCA(int a,int b){
	if(dep[a]<dep[b])swap(a,b);
	while(dep[a]>dep[b]){
		a=dp[a][lg[dep[a]-dep[b]]-1];
	}
	if(a==b)return a;
	for(int i=lg[dep[a]]-1;i>=0;i--){
		if(dp[a][i]!=dp[b][i]){
			a=dp[a][i];
			b=dp[b][i];
		}
	}
	return dp[a][0];
}
int work(int x,int y){
	if(x==y)return 1;
	int _lca=LCA(x,y);
	int d_rt=(dep[x]-dep[_lca])*2;
	int d_lf=dep[y]-dep[_lca];
	return d_rt+d_lf;
}
int d[105];
int main(){
	for(int i=1;i<=200;i++)lg[i]=lg[i-1]+(1<<lg[i-1]==1);
	n=read();
	for(int i=1;i<n;i++){
		int u=read(),v=read();
		node[u].push_back(v);
		node[v].push_back(u);
	}
	x=read(),y=read();
	dfs(1,0);
	int maxd=-1,maxw=-1;
	for(int i=1;i<=n;i++){
		d[dep[i]]++;
		maxd=max(maxd,dep[i]);
	}
	for(int i=1;i<=100;i++)maxw=max(maxw,d[i]);
	printf("%d\n%d\n%d",maxd,maxw,work(x,y));
	return 0;
}
2024/1/8 13:57
加载中...