Codeforces 601D Acyclic Organic Compounds（dfs+字典树合并）

最新推荐文章于 2021-09-27 19:49:24 发布

JeraKrs

最新推荐文章于 2021-09-27 19:49:24 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构-字典树 CF

本文链接：https://blog.youkuaiyun.com/keshuai19940722/article/details/50166635

CF 同时被 3 个专栏收录

248 篇文章

订阅专栏

GRADE:B

179 篇文章

订阅专栏

数据结构-字典树

31 篇文章

订阅专栏

本文提供了一种解决 Codeforces 601D 题目的算法思路，通过使用前缀树（Trie）来高效地处理字符串匹配问题，特别是在涉及大量字符串查询的场景中。该解决方案详细介绍了如何构建前缀树并利用其进行遍历和维护，以计算特定类型的有机化合物数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：Codeforces 601D Acyclic Organic Compounds

代码

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;
const int maxn = 300005;
const int SIGMA_SIZE = 26;
typedef long long ll;

int SZ = 0;
struct Node {
    int sz, ch[SIGMA_SIZE];
    Node () { memset(ch, 0, sizeof(ch)); }
}nd[maxn<<1];

int newNode () {
    ++SZ;
    memset(nd[SZ].ch, 0, sizeof(nd[SZ].ch));
    return SZ;
}

void maintain(int u) {
    nd[u].sz = 1;
    for (int i = 0; i < SIGMA_SIZE; i++) if (nd[u].ch[i])
        nd[u].sz += nd[nd[u].ch[i]].sz;
}

void merge(int &a, int b) {
    if (a == 0) {
        a = b;
    } else {
        for (int i = 0; i < SIGMA_SIZE; i++) {
            if (nd[b].ch[i])
                merge(nd[a].ch[i], nd[b].ch[i]);
        }
    }
    maintain(a);
}

int N;
ll C[maxn];
char S[maxn];
vector<int> G[maxn];

void init () {
    scanf("%d", &N);
    for (int i = 1; i <= N; i++) {
        G[i].clear();
        scanf("%lld", &C[i]);
    }
    scanf("%s", S + 1);

    int u, v;
    for (int i = 1; i < N; i++) {
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
}

int dfs(int u, int f) {
    int p = newNode();

    for (int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        if (v == f) continue;
        merge(p, dfs(v, u));
    }
    maintain(p);

    int r = newNode();
    nd[r].ch[S[u] - 'a'] = p;
    maintain(r);
    C[u] += nd[r].sz - 1;

    return r;
}

int main () {
    init();
    dfs(1, 0);
    ll ans = 0, cnt;
    for (int i = 1; i <= N; i++) {
        if (C[i] > ans) { ans = C[i]; cnt = 0; }
        if (C[i] == ans) { cnt++; }
    }
    printf("%lld\n%lld\n", ans, cnt);
    return 0;
}