LCA 16pts
查看原帖
LCA 16pts
323887
xieyuhao2022楼主2022/10/11 21:59
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct E{
	int to,nxt;
}edge[N<<1];
int head[N],tot;
int n,q,a[N],cnt[N][2],dep[N];
int f[N][20];
//0->H,1->G 
void add(int u,int v){
	edge[++tot].to=v;
	edge[tot].nxt=head[u];
	head[u]=tot;
}
void dfs(int u,int fa){
	cnt[u][a[u]]=cnt[fa][a[u]]+1;
	cnt[u][a[u]^1]=cnt[fa][a[u]^1];
	f[u][0]=fa;
	for(register int i=0;i<=19;i++){
		f[u][i+1]=f[f[u][i]][i];
	}
	for(register int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(v==fa) continue;
		dep[v]=dep[u]+1;
		dfs(v,u);
	}
}
int LCA(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	for(register int i=20;i>=0;i--){
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
	}
	if(x==y) return x;
	for(register int i=20;i>=0;i--){
		if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	}
	return f[x][0];
}
int main(){
	scanf("%d%d",&n,&q);
	for(register int i=1;i<=n;i++){
		char ch=getchar();
		while(ch<'A'||ch>'Z') ch=getchar();
		a[i]=(ch=='G');
	}
	for(register int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v),add(v,u);
	}
	dep[1]=1;
	dfs(1,0);
	while(q--){
		int x,y;
		scanf("%d%d",&x,&y);
		char ch=getchar();
		while(ch<'A'||ch>'Z') ch=getchar();
		int lca=LCA(x,y);
		int h=cnt[x][0]+cnt[y][0]-2*cnt[lca][0]+(a[lca]==0);
		int g=cnt[x][1]+cnt[y][1]-2*cnt[lca][1]+(a[lca]==1);
		if(ch=='H') {
			if(h) printf("1");
			else printf("0");
		}
		if(ch=='G') {
			if(g) printf("1");
			else printf("0");
		}
	}
	return 0;
} 
2022/10/11 21:59
加载中...