QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#624473#6729. Unrooted Triek1nsomAC ✓616ms104764kbC++173.9kb2024-10-09 15:55:332024-10-09 15:55:33

Judging History

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

  • [2024-10-09 15:55:33]
  • 评测
  • 测评结果:AC
  • 用时:616ms
  • 内存:104764kb
  • [2024-10-09 15:55:33]
  • 提交

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