QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#137715 | #6729. Unrooted Trie | Yarema# | AC ✓ | 264ms | 32792kb | C++17 | 1.8kb | 2023-08-10 17:01:35 | 2023-08-10 17:01:36 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define SZ(a) (int)a.size()
#define ALL(a) a.begin(), a.end()
#define FOR(i, a, b) for (int i = (a); i<(b); ++i)
#define RFOR(i, b, a) for (int i = (b)-1; i>=(a); --i)
#define MP make_pair
#define PB push_back
#define F first
#define S second
#define FILL(a, b) memset(a, b, sizeof(a))
typedef long long LL;
typedef pair<int, int> PII;
typedef vector<int> VI;
const int N = 100447;
LL a[N];
LL ans[N];
vector<PII> g[N];
void dfs(int v, int par = -1)
{
vector<int> cnt(26);
for (auto [to, c] : g[v])
{
if (to != par)
dfs(to, v);
cnt[c]++;
}
int two = -1;
FOR (i, 0, 26)
{
if (cnt[i] > 2)
{
a[0] = N;
return;
}
if (cnt[i] == 2)
{
if (two == -1) two = i;
else a[0] = N;
}
}
if (two == -1) return;
bool is_par = 0;
for (auto [to, c] : g[v]) if (to == par && c == two) is_par = 1;
if (is_par)
{
a[v]++;
for (auto [to, c] : g[v]) if (c == two && to != par) a[to]--;
}
else
{
a[0]++;
for (auto [to, c] : g[v]) if (c == two) a[to]--;
}
}
void dfs2(int v, int par = -1, LL x = 0)
{
x += a[v];
ans[v] = x;
for (auto [to, c] : g[v])
if (to != par)
dfs2(to, v, x);
}
void solve()
{
int n;
cin >> n;
FOR (i, 0, n - 1)
{
int u, v;
char ch;
cin >> u >> v >> ch;
u--, v--;
g[u].PB({v, ch - 'a'});
g[v].PB({u, ch - 'a'});
}
dfs(0);
dfs2(0);
int res = 0;
FOR (i, 0, n)
{
//cerr << i << ' ' << ans[i] << ' ' << a[i] << '\n';
if (ans[i] == 0)
res++;
}
cout << res << '\n';
FOR (i, 0, n)
{
g[i].clear();
a[i] = 0;
ans[i] = 0;
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int t;
cin >> t;
while (t--)
{
solve();
}
return 0;
}
这程序好像有点Bug,我给组数据试试?
详细
Test #1:
score: 100
Accepted
time: 1ms
memory: 5848kb
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: 264ms
memory: 32792kb
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