蒟蒻倍增LCA84分求帮,#7#8WA,死活调不好
查看原帖
蒟蒻倍增LCA84分求帮,#7#8WA,死活调不好
637073
wujingfey楼主2023/2/10 10:49
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m,dep[N],f[N][25];
char c,s[N];
int sumG[N],sumH[N];
vector<int> e[N];
void dfs(int u,int fa,int d){
	dep[u]=d;
	f[u][0]=fa;
	sumG[u]=sumG[fa]+(s[u]=='G');
	sumH[u]=sumH[fa]+(s[u]=='H');
	for(int i=1;i<=20;i++)
		f[u][i]=f[f[u][i-1]][i-1];
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];
		if(v!=fa) dfs(v,u,d+1);
	}
}
int LCA(int u,int v){
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=20;i>=0;i--)
		if(dep[u]-(1<<i)>=dep[v]) u=f[u][i];
	if(u==v) return u;
	for(int i=20;i>=0;i--)
		if(f[u][i]!=f[v][i]) u=f[u][i],v=f[u][i];
	return f[u][0];
}
int main(){
	scanf("%d%d",&n,&m);
	cin>>(s+1);
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(1,0,1);
	for(int i=1;i<=m;i++){
		int u,v,p;
		cin>>u>>v>>c;
		p=f[LCA(u,v)][0];
		if(c=='G') 
			cout<<(sumG[u]-sumG[p]>0 || sumG[v]-sumG[p]>0);
		else 
			cout<<(sumH[u]-sumH[p]>0 || sumH[v]-sumH[p]>0);
	}
	return 0;
}

2023/2/10 10:49
加载中...