UOJ Logo

NOI.AC

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#214863#2642. colorKXG100863ms15600kbC++112.0kb2024-11-22 21:34:042024-11-22 23:13:52

answer

#include <stdio.h>
#include <vector>
#include <cstring>
#include <cmath>
using namespace std;
int n, m;
char colors[100010];
vector<int> G[100010];
int depth[100010], fa[20][100010], cntH[100010], cntG[100010];
void dfs(int u, int p) {
    fa[0][u] = p;
    depth[u] = depth[p] + 1;
    cntH[u] = cntH[p] + (colors[u] == 'H' ? 1 : 0);
    cntG[u] = cntG[p] + (colors[u] == 'G' ? 1 : 0);
    for (int v : G[u]) {
        if (v != p) {
            dfs(v, u);
        }
    }
}
void init() {
    for (int k = 1; k < 20; k++) {
        for (int u = 1; u <= n; u++) {
            if (fa[k - 1][u] != -1) {
                fa[k][u] = fa[k - 1][fa[k - 1][u]];
            } else {
                fa[k][u] = -1;
            }
        }
    }
}
int lca(int x, int y) {
    if (depth[x] < depth[y]) swap(x, y);
    for (int k = 19; k >= 0; k--) {
        if (depth[fa[k][x]] >= depth[y]) {
            x = fa[k][x];
        }
    }
    if (x == y) return x;
    for (int k = 19; k >= 0; k--) {
        if (fa[k][x] != fa[k][y]) {
            x = fa[k][x];
            y = fa[k][y];
        }
    }
    return fa[0][x];
}
int main() {
    scanf("%d%d", &n, &m);
    scanf("%s", colors + 1);
    for (int i = 1; i < n; i++) {
        int x, y;
        scanf("%d%d", &x, &y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    memset(fa, -1, sizeof(fa));
    dfs(1, 0);
    init();
    for (int i = 0; i < m; i++) {
        int x, y;
        char t[10];
        scanf("%d%d%s", &x, &y, t);
        int ancestor = lca(x, y);
        int count_t;
        if (t[0] == 'H') {
            count_t = cntH[x] + cntH[y] - 2 * cntH[ancestor] + (colors[ancestor] == 'H' ? 1 : 0);
        } else {
            count_t = cntG[x] + cntG[y] - 2 * cntG[ancestor] + (colors[ancestor] == 'G' ? 1 : 0);
        }
        if (count_t > 0) {
            putchar('1');
        } else {
            putchar('0');
        }
    }
    return 0;
}

详细

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

Test #1:

score: 10
Accepted
time: 3ms
memory: 11280kb

input:

920 900
HHHHHHHHHHHHHHHHHHHHHHHHHGHHHHHHHHHHHHGHHHHHHHHHHHHHHHHHHHHHHHGHHGHHHHHHHGGHGHHHGHHGHGHHHHHH...

output:

0010100111100111101010001000010101100010001110110110001101100101111000101111100001111001110100001111...

result:

ok single line: '001010011110011110101000100001...1010100111010011101110111010101'

Test #2:

score: 10
Accepted
time: 7ms
memory: 11284kb

input:

927 949
HHHHGHGGGHHHHGHGHHHHHGHGGHGGGHHHGGHHHHHHGGGGGGHHHGGHHHHGHHHHGGGHHHGHGHHHHGGGGHHGHHGGHGHGGGGG...

output:

1111000011110111101101100110110011110101011000110111111111101110110011111011111110110011101010111111...

result:

ok single line: '111100001111011110110110011011...0011001011010111111011111001110'

Test #3:

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

input:

934 998
HHHHGHHHGHHGGGHGGHGHGHHHGHHGHGGGHHHHHGHHGGHHHHHHGHHHGGHHHHHGGHHHGHHHHHHHGHHGHGHGGHHGHGHHHGGH...

output:

1101010111111010110101011111110111111101101100011111001101110111111111011111101001111011001010101111...

result:

ok single line: '110101011111101011010101111111...1011101110011100101111100101110'

Test #4:

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

input:

941 947
HHHHHHHHHHHHHHHGHHHGHHHGHHHHGHHHGHHHHHHHHGHHHHHHHHHGHHHHHGHHHHHGHHGHHHHGHGGHHGHHHHHHHHHHGGHH...

output:

1111111011010110001101010101110100110001000101100101111101110001111111111110111011010101001111101111...

result:

ok single line: '111111101101011000110101010111...0101100011110110011000000000011'

Test #5:

score: 10
Accepted
time: 137ms
memory: 15396kb

input:

92189 98896
HHHHHHHHHGHHHHGGGHGHHHHHHGHHHHHGHHHHHHHHHGHGHHHGHHHHHHGGHHHGHGGHHHHHHHHHGHHHHHHGHGHGHHHG...

output:

1110111111111111111111111111111111111011110111111111111111110111111101111111111111111111111111101111...

result:

ok single line: '111011111111111111111111111111...1111111111111111111111111111111'

Test #6:

score: 10
Accepted
time: 146ms
memory: 15560kb

input:

95803 95747
HHHHHHHHHHHHHGHHHHHHHHHHHHHHHHHHGHHGHGHHHHGHGHHHHHHGGHHHHHHHHHHHHGHHHGHHHHHHHHHGHGGHHHHH...

output:

1011011101111111111111111101111111111011101101111110111111101111111011110111111101111111001100101111...

result:

ok single line: '101101110111111111111111110111...1111101101111110111111111001111'

Test #7:

score: 10
Accepted
time: 144ms
memory: 15416kb

input:

92610 90996
HHGGGHHGHGGGHGHGHGGGGGGHGGGHGGHGHGHGGHHHHHGHGGHGHGGGGHGGGHGGHHGHHHGGHGHGGHGGGHGGGGGGGGGG...

output:

1111111111111111111011111111111111111110111111111111100111111111011111111111101101111111111111011111...

result:

ok single line: '111111111111111111101111111111...1111110011111111111111111111111'

Test #8:

score: 10
Accepted
time: 131ms
memory: 15580kb

input:

96224 91494
HHHHHHHHGGHHGHHHHHHHHHHHHHGHHHGHHHHHHHHHHHHHHHHHGGHHHHHHGHHHHHHHHHHHHHHHHHGHGGGHGHGHHHHH...

output:

1111010110111100111111111111011011010101011111011101111011111011111011111111111111111111111011111101...

result:

ok single line: '111101011011110011111111111101...1101011011111111111111101101111'

Test #9:

score: 10
Accepted
time: 140ms
memory: 15432kb

input:

93031 96743
HHHHHHHHHGHHHHHHHHHHHHHHHHHGGHHHHGHHHGHHHHHHHHHGHHHHHHHHHHHHHHGGHHHHHGGHGHHHGHHHGHHHHHHG...

output:

0100011111101111011111110111001111101111110101110110111111111111111101111110110001011110011111011111...

result:

ok single line: '010001111110111101111111011100...1111111101110011111101011111110'

Test #10:

score: 10
Accepted
time: 155ms
memory: 15600kb

input:

96645 93594
HHHHHHHHHGHGHHHHHHHGHHHHGGGHHHHHHHHHHHHHHHHHHGHHHHHGHGHHHHHGHGHHHHHGHHHHGHHHHGGHHGHHHHGH...

output:

0101000110111110011111111111110111101011011101111110111111011011111111111110111110101111111111111111...

result:

ok single line: '010100011011111001111111111111...1110011111111111110110011111111'