QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#351038 | #6729. Unrooted Trie | Dream_56 | AC ✓ | 259ms | 20660kb | C++14 | 2.3kb | 2024-03-11 13:14:06 | 2024-03-11 13:14:06 |
Judging History
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