QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#624473 | #6729. Unrooted Trie | k1nsom | AC ✓ | 616ms | 104764kb | C++17 | 3.9kb | 2024-10-09 15:55:33 | 2024-10-09 15:55:33 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
const int N = 1e5 + 5;
struct node
{
int val, tag;
} tr[N * 20];
int n;
vector<int> e[N][26];
void build(int p, int l, int r) // 建树
{
if (l == r)
{
tr[p].val = 0;
tr[p].tag = 0;
return;
}
int mid = (l + r) / 2;
build(2 * p, l, mid);
build(2 * p + 1, mid + 1, r);
tr[p].val = tr[2 * p].val + tr[2 * p + 1].val;
tr[p].tag = 0;
}
void push_down(int p, int len)
{
// cout << p << endl;
tr[p * 2].tag += tr[p].tag;
tr[p * 2 + 1].tag += tr[p].tag;
tr[p * 2].val = tr[2 * p].val + tr[p].tag * (len - len / 2);
tr[p * 2 + 1].val = tr[p * 2 + 1].val + tr[p].tag * (len / 2); // 右边的区间可能要短一点
tr[p].tag = 0;
}
void update(int l, int r, int p, int s, int t, int val) // l,r修改区间;s,t查询区间
{
// cout << s << ' ' << t << endl;
if (t < l || s > r)
return;
if (l <= s && t <= r)
{
tr[p].val = tr[p].val + (t - s + 1) * val;
if (s != t)
tr[p].tag += val;
return;
}
push_down(p, t - s + 1);
int mid = (s + t) >> 1;
update(l, r, 2 * p, s, mid, val);
update(l, r, 2 * p + 1, mid + 1, t, val);
tr[p].val = tr[p * 2].val + tr[p * 2 + 1].val;
}
int query(int l, int r, int p, int s, int t)
{
if (t < l || s > r)
return 0;
if (l <= s && t <= r)
return tr[p].val;
push_down(p, t - s + 1);
int mid = (s + t) / 2, ans = 0;
return query(l, r, 2 * p, s, mid) + query(l, r, 2 * p + 1, mid + 1, t);
}
int dfsn, dfn[N], ls[N], rs[N], f[N];
void dfs(int u, int fa)
{
f[u] = fa;
dfn[u] = rs[u] = ls[u] = ++dfsn;
for (int j = 0; j < 26; j++)
for (auto to : e[u][j])
{
if (to != fa)
{
dfs(to, u);
ls[u] = min(ls[u], ls[to]);
rs[u] = max(rs[u], rs[to]);
}
}
}
void solve()
{
cin >> n;
for (int i = 1; i <= n; i++)
for (int j = 0; j < 26; j++)
e[i][j].clear();
for (int i = 1; i < n; i++)
{
int x, y;
char c;
cin >> x >> y >> c;
e[x][c - 'a'].push_back(y);
e[y][c - 'a'].push_back(x);
}
bool ok = 0;
for (int i = 1; i <= n; i++)
{
int cnt = 0;
for (int j = 0; j < 26; j++)
{
if (e[i][j].size() >= 3)
{
ok = 1;
}
else if (e[i][j].size() == 2)
cnt++;
}
if (cnt >= 2)
ok = 1;
}
if (ok)
{
cout << 0 << endl;
return;
}
build(1, 1, n);
dfsn = 0;
dfs(1, 0);
for (int i = 1; i <= n; i++)
{
for (int j = 0; j < 26; j++)
{
if (e[i][j].size() == 2)
{
for (int k = 0; k < 26; k++)
{
if (k != j)
{
for (auto to : e[i][k])
{
if (to != f[i])
update(ls[to], rs[to], 1, 1, n, 1);
else
{
update(1, n, 1, 1, n, 1);
update(ls[i], rs[i], 1, 1, n, -1);
}
}
}
}
update(dfn[i], dfn[i], 1, 1, n, 1);
}
}
}
int ans = 0;
for (int i = 1; i <= n; i++)
{
if (query(i, i, 1, 1, n) == 0)
ans++;
}
cout << ans << endl;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T = 1;
cin >> T;
while (T--)
{
solve();
}
return 0;
}
这程序好像有点Bug,我给组数据试试?
详细
Test #1:
score: 100
Accepted
time: 10ms
memory: 69112kb
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: 616ms
memory: 104764kb
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