kuangbin字典树

最新推荐文章于 2025-12-02 21:05:32 发布

原创最新推荐文章于 2025-12-02 21:05:32 发布 · 141 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #c语言 #算法

算法竞赛专栏收录该内容

20 篇文章

订阅专栏

该博客介绍了一种利用字典树（Trie）解决找出字符串唯一前缀的方法。通过遍历输入字符串并更新字典树中每个节点的计数，当遇到某个节点的计数为1时，表明其为唯一的前缀。代码实现包括插入字符串到字典树和查询唯一前缀的函数，最后展示了应用示例。

POJ - 2001 Shortest Prefixes

思路

在字典树标记字母出现的次数即可
当发现cnt[p] == 1时，说明这个字母前缀只出现过一次，直接返回即可

代码

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 1010 * 26;
int son[N][26];
int idx;
char str[N][26];
int cnt[N];


void insert(char str[]) {
    int p = 0;
    for (int i = 0; str[i] != '\0'; i ++) {
        int u = str[i] - 'a';
        if(!son[p][u])
            son[p][u] = ++idx;
        p = son[p][u];
        cnt[p]++;
    }
    
}

string query(char str[]) {
    int p = 0;
    string s = "";
    for (int i = 0; str[i] != '\0'; i ++) {
        int u = str[i] - 'a';
        p = son[p][u];
        if(cnt[p] == 1){
            s += str[i];
            return s;
        }
        else {
            s += str[i];
        }
    }

    return s;
}

int main() {
    int n = 0;
    while(cin >> str[n]) {
        insert(str[n]);
        n++;
    }

    for (int i = 0; i < n; i ++) {
        cout << str[i] << ' ' << query(str[i]) << endl;
    }

    return 0;
}