LCA全RE,本地没事。。
查看原帖
LCA全RE,本地没事。。
469492
Vergil_Sparda楼主2022/10/13 11:26

为什么啊

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m;
string s;
struct edge{
	int to,next;
}e[1000001];
int head[1000001],tot;
int f[1000001][30],v[1000001][30];
int lg[1000001];
int dep[1000001];
int a[1000001];
void add(int x,int y){
	tot++;
	e[tot].to=y;
	e[tot].next=head[x];
	head[x]=tot;
}
void dfs(int id,int fa){
	f[id][0]=fa;
	dep[id]=dep[fa]+1;
	v[id][0]=a[id]|a[fa];
	for(int i=1;i<=lg[dep[id]];i++){
		f[id][i]=f[f[id][i-1]][i-1];
		v[id][i]=v[f[id][i-1]][i-1]|v[id][i-1];
	}
	for(int i=head[i];i;i=e[i].next){
		if(e[i].to!=fa)
		dfs(e[i].to,id);
	}
}
int lca(int x,int y){
	int ans=0;
	if(dep[x]<dep[y]){
		swap(x,y);
	}
	ans|=a[x];
	ans|=a[y];
	while(dep[x]>dep[y]){
		ans|=v[x][lg[dep[x]-dep[y]]-1];
		x=f[x][lg[dep[x]-dep[y]]-1];
	}
	if(x==y){
		return ans;
	}
	for(int i=lg[dep[x]]-1;i>=0;i--){
		if(f[x][i]!=f[y][i]){
			ans|=v[x][i];
			ans|=v[y][i];
			x=f[x][i];
			y=f[y][i];
		}
	}
	ans|=v[x][0];
	ans|=v[y][0];
	return ans;
}
string ANS;
int main(){
	cin>>n>>m;
	cin>>s;
	s=" "+s;
	for(int i=1;i<s.size();i++){
		if(s[i]=='G'){
			a[i]=1;
		}
		else{
			a[i]=2;
		}
	}
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	for(int i=1;i<=n;i++){
		lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	}
	dfs(1,0);
	for(int i=1;i<=m;i++){
		int x,y,z;
		char c;
		cin>>x>>y>>c;
		if(c=='G')z=1;
		else z=2;
		if((lca(x,y)&z)==z){
			ANS+="1";
		}
		else{
			ANS+="0";
		}
	}
	cout<<ANS;
	// for(int i=1;i<=n;i++){
	// 	cout<<i<<" "<<lg[i]<<endl;
	// }
	return 0;
}
2022/10/13 11:26
加载中...