UOJ Logo

NOI.AC

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#191220#2642. colortkswls100447ms22800kbC++111.9kb2023-10-08 10:21:372023-10-08 12:40:07

answer

#include<bits/stdc++.h>
using namespace std;
struct edge {
	int v, nxt;
} a[200005];
int n, m, fa[100005][25], s, cnt, d[100005], sum[100005][25], num[100005], ans, dis;
int head[100005];
void add(int p, int q) {
	a[++cnt].nxt = head[p];
	head[p] = cnt;
	a[cnt].v = q;
}
void dfs(int q) {
	sum[q][0] = num[q];
	for (int i = 1; (1 << i) + 1 <= d[q]; i++) {
		fa[q][i] = fa[fa[q][i - 1]][i - 1];
		sum[q][i] = sum[q][i - 1] + sum[fa[q][i - 1]][i - 1];
	}
	for (int i = head[q]; i; i = a[i].nxt) {
		if (d[a[i].v]) continue;
		fa[a[i].v][0] = q;
		d[a[i].v] = d[q] + 1;
		dfs(a[i].v);
	}
}
int lca(int p, int q) {
	if (d[q] > d[p]) swap(p, q);
	ans = dis = 0;
//	cout << p << " " << q << " " << dis << ' ' << ans << d[p] << ' ' << d[q] << endl;
	for (int i = 20; i >= 0; i--) {
		if (d[p] - (1 << i) >= d[q]) {
			ans += sum[p][i];
			p = fa[p][i];
			dis += (1 << i);
		}
	}
//	cout << p << " " << q << " " << dis << ' ' << ans << endl;
	dis++;
	ans += sum[p][0];
	if (p == q) return p;
//	cout << p << " " << q << " " << dis << ' ' << ans << endl;
	for (int i = 20; i >= 0; i--) {
		if (fa[q][i] != fa[p][i]) {
			ans += sum[q][i] + sum[p][i];
			q = fa[q][i];
			dis += 2 * (1 << i);
			p = fa[p][i];
		}
	}
	dis += 3;
	ans += sum[p][0] + sum[q][0] + sum[fa[p][0]][0];
	return fa[p][0];
}
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin >> n >> m;
	int p, q;
	char c;
	for (int i = 1; i <= n; i++) {
		cin >> c;
		if (c == 'G') num[i] = 1;
	}
	for (int i = 1; i < n; i++) {
		cin >> p >> q;
		add(p, q);
		add(q, p);
	}
	d[1] = 1;
	dfs(1);
	for (int i = 1; i <= m; i++) {
		cin >> p >> q >> c;
		lca(p, q);
//		cout << ans << "|" << dis << endl;
		if (c == 'H') {
			if (dis == ans) {
				cout << 0;
			} else {
				cout << 1;
			}
		} else {
			if (ans == 0) {
				cout << 0;
			} else {
				cout << 1;
			}
		}
	}
	return 0;
}

详细

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

Test #1:

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

input:

920 900
HHHHHHHHHHHHHHHHHHHHHHHHHGHHHHHHHHHHHHGHHHHHHHHHHHHHHHHHHHHHHHGHHGHHHHHHHGGHGHHHGHHGHGHHHHHH...

output:

0010100111100111101010001000010101100010001110110110001101100101111000101111100001111001110100001111...

result:

ok single line: '001010011110011110101000100001...1010100111010011101110111010101'

Test #2:

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

input:

927 949
HHHHGHGGGHHHHGHGHHHHHGHGGHGGGHHHGGHHHHHHGGGGGGHHHGGHHHHGHHHHGGGHHHGHGHHHHGGGGHHGHHGGHGHGGGGG...

output:

1111000011110111101101100110110011110101011000110111111111101110110011111011111110110011101010111111...

result:

ok single line: '111100001111011110110110011011...0011001011010111111011111001110'

Test #3:

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

input:

934 998
HHHHGHHHGHHGGGHGGHGHGHHHGHHGHGGGHHHHHGHHGGHHHHHHGHHHGGHHHHHGGHHHGHHHHHHHGHHGHGHGGHHGHGHHHGGH...

output:

1101010111111010110101011111110111111101101100011111001101110111111111011111101001111011001010101111...

result:

ok single line: '110101011111101011010101111111...1011101110011100101111100101110'

Test #4:

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

input:

941 947
HHHHHHHHHHHHHHHGHHHGHHHGHHHHGHHHGHHHHHHHHGHHHHHHHHHGHHHHHGHHHHHGHHGHHHHGHGGHHGHHHHHHHHHHGGHH...

output:

1111111011010110001101010101110100110001000101100101111101110001111111111110111011010101001111101111...

result:

ok single line: '111111101101011000110101010111...0101100011110110011000000000011'

Test #5:

score: 10
Accepted
time: 74ms
memory: 21808kb

input:

92189 98896
HHHHHHHHHGHHHHGGGHGHHHHHHGHHHHHGHHHHHHHHHGHGHHHGHHHHHHGGHHHGHGGHHHHHHHHHGHHHHHHGHGHGHHHG...

output:

1110111111111111111111111111111111111011110111111111111111110111111101111111111111111111111111101111...

result:

ok single line: '111011111111111111111111111111...1111111111111111111111111111111'

Test #6:

score: 10
Accepted
time: 73ms
memory: 22608kb

input:

95803 95747
HHHHHHHHHHHHHGHHHHHHHHHHHHHHHHHHGHHGHGHHHHGHGHHHHHHGGHHHHHHHHHHHHGHHHGHHHHHHHHHGHGGHHHHH...

output:

1011011101111111111111111101111111111011101101111110111111101111111011110111111101111111001100101111...

result:

ok single line: '101101110111111111111111110111...1111101101111110111111111001111'

Test #7:

score: 10
Accepted
time: 68ms
memory: 21900kb

input:

92610 90996
HHGGGHHGHGGGHGHGHGGGGGGHGGGHGGHGHGHGGHHHHHGHGGHGHGGGGHGGGHGGHHGHHHGGHGHGGHGGGHGGGGGGGGGG...

output:

1111111111111111111011111111111111111110111111111111100111111111011111111111101101111111111111011111...

result:

ok single line: '111111111111111111101111111111...1111110011111111111111111111111'

Test #8:

score: 10
Accepted
time: 70ms
memory: 22704kb

input:

96224 91494
HHHHHHHHGGHHGHHHHHHHHHHHHHGHHHGHHHHHHHHHHHHHHHHHGGHHHHHHGHHHHHHHHHHHHHHHHHGHGGGHGHGHHHHH...

output:

1111010110111100111111111111011011010101011111011101111011111011111011111111111111111111111011111101...

result:

ok single line: '111101011011110011111111111101...1101011011111111111111101101111'

Test #9:

score: 10
Accepted
time: 78ms
memory: 21992kb

input:

93031 96743
HHHHHHHHHGHHHHHHHHHHHHHHHHHGGHHHHGHHHGHHHHHHHHHGHHHHHHHHHHHHHHGGHHHHHGGHGHHHGHHHGHHHHHHG...

output:

0100011111101111011111110111001111101111110101110110111111111111111101111110110001011110011111011111...

result:

ok single line: '010001111110111101111111011100...1111111101110011111101011111110'

Test #10:

score: 10
Accepted
time: 84ms
memory: 22800kb

input:

96645 93594
HHHHHHHHHGHGHHHHHHHGHHHHGGGHHHHHHHHHHHHHHHHHHGHHHHHGHGHHHHHGHGHHHHHGHHHHGHHHHGGHHGHHHHGH...

output:

0101000110111110011111111111110111101011011101111110111111011011111111111110111110101111111111111111...

result:

ok single line: '010100011011111001111111111111...1110011111111111110110011111111'