QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#351038#6729. Unrooted TrieDream_56AC ✓259ms20660kbC++142.3kb2024-03-11 13:14:062024-03-11 13:14:06

Judging History

你现在查看的是最新测评结果

  • [2024-03-11 13:14:06]
  • 评测
  • 测评结果:AC
  • 用时:259ms
  • 内存:20660kb
  • [2024-03-11 13:14:06]
  • 提交

answer

#include <bits/stdc++.h>
#define MAXN ((int) 1e5)
using namespace std;

int n, ans;

vector<int> e[MAXN + 10];
vector<char> v[MAXN + 10];

int clk, bgn[MAXN + 10], fin[MAXN + 10];
int f[MAXN + 10];

void dfs(int sn, int fa) {
    bgn[sn] = ++clk;
    for (int fn : e[sn]) if (fn != fa) dfs(fn, sn);
    fin[sn] = clk;
}

void solve() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) e[i].clear(), v[i].clear();
    for (int i = 1; i < n; i++) {
        int x, y;
        char z[3];
        scanf("%d%d%s", &x, &y, z);
        e[x].push_back(y); v[x].push_back(z[0]);
        e[y].push_back(x); v[y].push_back(z[0]);
    }

    clk = 0; dfs(1, 0);
    memset(f, 0, sizeof(int) * (n + 3));
    for (int sn = 1; sn <= n; sn++) {
        vector<int> vec[26];
        for (int i = 0; i < e[sn].size(); i++) vec[v[sn][i] - 'a'].push_back(e[sn][i]);
        int t = -1;
        for (int i = 0; i < 26; i++) {
            // 相同的字母有至少三个
            if (vec[i].size() > 2) { printf("0\n"); return; }
            else if (vec[i].size() == 2) {
                // 有至少两对相同的字母
                if (t >= 0) { printf("0\n"); return; }
                t = i;
            }
        }
        // 没有相同字母,不影响答案
        if (t == -1) continue;

        int x = vec[t][0], y = vec[t][1];
        if (bgn[x] > bgn[y]) swap(x, y);
        if (bgn[sn] < bgn[x] && bgn[sn] < bgn[y]) {
            // sn 在 dfs 中是 x 和 y 的父节点
            // 只有 x 的子树和 y 的子树可能成为 trie 的根
            f[1]++; f[n + 1]--;
            f[bgn[x]]--; f[fin[x] + 1]++;
            f[bgn[y]]--; f[fin[y] + 1]++;
        } else {
            // x 在 dfs 中是 sn 的父节点
            // 除了 sn 的子树以外的地方可能成为 trie 的根
            f[bgn[sn]]++; f[fin[sn] + 1]--;
            // y 的子树可能成为 trie 的根
            f[bgn[y]]--; f[fin[y] + 1]++;
        }
    }

    ans = 0;
    for (int i = 1; i <= n; i++) {
        f[i] += f[i - 1];
        // 没有标记的位置可以成为 trie 的根
        if (f[i] == 0) ans++;
    }
    printf("%d\n", ans);
}

int main() {
    int tcase; scanf("%d", &tcase);
    while (tcase--) solve();
    return 0;
}

这程序好像有点Bug,我给组数据试试?

详细

Test #1:

score: 100
Accepted
time: 2ms
memory: 8560kb

input:

2
6
3 1 a
3 2 a
3 4 b
4 5 c
4 6 d
6
3 1 a
3 2 a
3 4 b
5 4 c
6 4 c

output:

2
0

result:

ok 2 number(s): "2 0"

Test #2:

score: 0
Accepted
time: 259ms
memory: 20660kb

input:

1112
19
15 18 a
7 18 c
11 14 b
10 17 b
8 14 a
1 3 b
12 2 a
16 3 c
16 4 b
2 3 a
15 5 a
3 18 d
16 9 a
18 13 b
8 4 b
17 7 a
9 6 a
13 19 a
52
8 32 a
14 51 a
30 52 a
48 36 b
27 39 c
39 51 b
35 15 a
51 52 d
45 51 e
39 26 a
20 12 b
34 18 a
9 12 e
25 5 a
9 13 b
41 51 c
1 52 n
33 14 c
22 30 b
17 4 b
12 52 c
...

output:

15
49
22
68
34
17
28
27
3
4
34
70
37
39
19
24
58
8
16
14
10
73
73
65
35
45
33
81
46
35
78
49
22
13
26
10
33
48
47
3
9
50
8
37
15
84
23
75
26
35
35
61
65
58
30
56
11
8
39
60
88
40
56
17
42
62
12
11
2
59
22
54
14
91
87
1
80
11
45
69
80
33
87
46
56
62
54
80
7
4
48
20
55
19
9
4
38
39
89
35
63
46
24
7
52...

result:

ok 1112 numbers