UOJ Logo

NOI.AC

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#191212#2642. colorsnow_trace1001613ms51844kbC++111.8kb2023-10-08 08:19:182023-10-08 12:39:22

answer

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m;
int col[200005],dep[200005];
vector<int>p[200005];
int lca[100005][18],ok1[100005][18],ok2[100005][18];
void dfs(int now,int fa){
	lca[now][0] = fa;dep[now] = dep[fa]+1;
	ok1[now][0] = (col[now] == 1)|(col[fa] ==1);
	ok2[now][0] = (col[now] == 2)|(col[fa] ==2);
	for(int i = 1;i<18;i++){
		lca[now][i] = lca[lca[now][i-1]][i-1];
		ok1[now][i] = ok1[now][i-1]|ok1[lca[now][i-1]][i-1];
		ok2[now][i] = ok2[now][i-1]|ok2[lca[now][i-1]][i-1];
	}
	for(int i =0;i<p[now].size();i++){
		if(p[now][i] == fa)continue;
		dfs(p[now][i],now);
	}
}bool query1(int x,int y){
	if(x == y)return col[x] == 1;
	int ok = 0;
	if(dep[x]<dep[y])swap(x,y);
	int dis = dep[x]-dep[y];
	for(int i = 0;dis;dis>>=1,i++){
		if(dis&1)ok|=ok1[x][i],x = lca[x][i];
	}if(x == y)return ok;
	for(int i = 17;i>=0;i--){
		if(lca[y][i]!=lca[x][i]){
			ok|=ok1[x][i]|ok1[y][i];
			x = lca[x][i],y = lca[y][i];
		}
	}return ok|ok1[x][0]|ok1[y][0];
}bool query2(int x,int y){
	if(x == y)return col[x] == 2;
	int ok = 0;
	if(dep[x]<dep[y])swap(x,y);
	int dis = dep[x]-dep[y];
	for(int i = 0;dis;dis>>=1,i++){
		if(dis&1)ok|=ok2[x][i],x = lca[x][i];
	}if(x == y)return ok;
	for(int i = 17;i>=0;i--){
		if(lca[y][i]!=lca[x][i]){
			ok|=ok2[x][i]|ok2[y][i];
			x = lca[x][i],y = lca[y][i];
		}
	}return ok|ok2[x][0]|ok2[y][0];
}
signed main(){
	cin >> n >> m;
	for(int i = 1;i<=n;i++){
		char c;cin >> c;
		if(c == 'H')col[i] = 1;
		else col[i] = 2;
	}for(int i = 1;i<n;i++){
		int x,y;cin >> x >> y;
		p[x].push_back(y),p[y].push_back(x);
	}dfs(1,1);
	for(int i = 1;i<=m;i++){
		int x,y;char c;cin>>  x>> y>>c;
		if(c == 'H'){
			int res = query1(x,y);
			if(res)cout << 1;
			else cout << 0;
		}else{
			int res = query2(x,y);
			if(res)cout << 1;
			else cout << 0;
		}
	}
	return 0;
}

Details

小提示:点击横条可展开更详细的信息

Test #1:

score: 10
Accepted
time: 2ms
memory: 6392kb

input:

920 900
HHHHHHHHHHHHHHHHHHHHHHHHHGHHHHHHHHHHHHGHHHHHHHHHHHHHHHHHHHHHHHGHHGHHHHHHHGGHGHHHGHHGHGHHHHHH...

output:

0010100111100111101010001000010101100010001110110110001101100101111000101111100001111001110100001111...

result:

ok single line: '001010011110011110101000100001...1010100111010011101110111010101'

Test #2:

score: 10
Accepted
time: 0ms
memory: 6392kb

input:

927 949
HHHHGHGGGHHHHGHGHHHHHGHGGHGGGHHHGGHHHHHHGGGGGGHHHGGHHHHGHHHHGGGHHHGHGHHHHGGGGHHGHHGGHGHGGGGG...

output:

1111000011110111101101100110110011110101011000110111111111101110110011111011111110110011101010111111...

result:

ok single line: '111100001111011110110110011011...0011001011010111111011111001110'

Test #3:

score: 10
Accepted
time: 6ms
memory: 6396kb

input:

934 998
HHHHGHHHGHHGGGHGGHGHGHHHGHHGHGGGHHHHHGHHGGHHHHHHGHHHGGHHHHHGGHHHGHHHHHHHGHHGHGHGGHHGHGHHHGGH...

output:

1101010111111010110101011111110111111101101100011111001101110111111111011111101001111011001010101111...

result:

ok single line: '110101011111101011010101111111...1011101110011100101111100101110'

Test #4:

score: 10
Accepted
time: 2ms
memory: 6404kb

input:

941 947
HHHHHHHHHHHHHHHGHHHGHHHGHHHHGHHHGHHHHHHHHGHHHHHHHHHGHHHHHGHHHHHGHHGHHHHGHGGHHGHHHHHHHHHHGGHH...

output:

1111111011010110001101010101110100110001000101100101111101110001111111111110111011010101001111101111...

result:

ok single line: '111111101101011000110101010111...0101100011110110011000000000011'

Test #5:

score: 10
Accepted
time: 278ms
memory: 49732kb

input:

92189 98896
HHHHHHHHHGHHHHGGGHGHHHHHHGHHHHHGHHHHHHHHHGHGHHHGHHHHHHGGHHHGHGGHHHHHHHHHGHHHHHHGHGHGHHHG...

output:

1110111111111111111111111111111111111011110111111111111111110111111101111111111111111111111111101111...

result:

ok single line: '111011111111111111111111111111...1111111111111111111111111111111'

Test #6:

score: 10
Accepted
time: 277ms
memory: 51448kb

input:

95803 95747
HHHHHHHHHHHHHGHHHHHHHHHHHHHHHHHHGHHGHGHHHHGHGHHHHHHGGHHHHHHHHHHHHGHHHGHHHHHHHHHGHGGHHHHH...

output:

1011011101111111111111111101111111111011101101111110111111101111111011110111111101111111001100101111...

result:

ok single line: '101101110111111111111111110111...1111101101111110111111111001111'

Test #7:

score: 10
Accepted
time: 257ms
memory: 49936kb

input:

92610 90996
HHGGGHHGHGGGHGHGHGGGGGGHGGGHGGHGHGHGGHHHHHGHGGHGHGGGGHGGGHGGHHGHHHGGHGHGGHGGGHGGGGGGGGGG...

output:

1111111111111111111011111111111111111110111111111111100111111111011111111111101101111111111111011111...

result:

ok single line: '111111111111111111101111111111...1111110011111111111111111111111'

Test #8:

score: 10
Accepted
time: 253ms
memory: 51648kb

input:

96224 91494
HHHHHHHHGGHHGHHHHHHHHHHHHHGHHHGHHHHHHHHHHHHHHHHHGGHHHHHHGHHHHHHHHHHHHHHHHHGHGGGHGHGHHHHH...

output:

1111010110111100111111111111011011010101011111011101111011111011111011111111111111111111111011111101...

result:

ok single line: '111101011011110011111111111101...1101011011111111111111101101111'

Test #9:

score: 10
Accepted
time: 287ms
memory: 50132kb

input:

93031 96743
HHHHHHHHHGHHHHHHHHHHHHHHHHHGGHHHHGHHHGHHHHHHHHHGHHHHHHHHHHHHHHGGHHHHHGGHGHHHGHHHGHHHHHHG...

output:

0100011111101111011111110111001111101111110101110110111111111111111101111110110001011110011111011111...

result:

ok single line: '010001111110111101111111011100...1111111101110011111101011111110'

Test #10:

score: 10
Accepted
time: 251ms
memory: 51844kb

input:

96645 93594
HHHHHHHHHGHGHHHHHHHGHHHHGGGHHHHHHHHHHHHHHHHHHGHHHHHGHGHHHHHGHGHHHHHGHHHHGHHHHGGHHGHHHHGH...

output:

0101000110111110011111111111110111101011011101111110111111011011111111111110111110101111111111111111...

result:

ok single line: '010100011011111001111111111111...1110011111111111110110011111111'