字典树

最新推荐文章于 2025-07-04 09:42:48 发布

原创最新推荐文章于 2025-07-04 09:42:48 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

ACM_数据结构基础专栏收录该内容

49 篇文章

订阅专栏

本文介绍了一种使用字典树（Trie）的数据结构来高效地存储和检索字符串集合的方法。通过插入操作将字符串加入到字典树中，并通过查找操作统计特定前缀出现的次数。该方法适用于文本搜索等应用场景。

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

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;

const int kind = 26;
struct node{
    int num;
    bool terminal;
    node *next[26];
    node(){
        num = 1;
        terminal = false;
        memset(next, 0, sizeof(next));
    }
};

string s1, s2;

void Insert(node *root, string s1){
    node *p = root;
    int index;
    int len = s1.length();
    for(int i = 0; i < len; i++){
        index = s1[i] - 'a';
        if(p -> next[index] == NULL){
            p -> next[index] = new node();
        }
        else{
            p -> next[index] -> num++;
        }
        p = p -> next[index];
    }
    p ->terminal = true;
}

int Find(node *root, string s2){
    node *p = root;
    int i, index;
    int len = s2.length();
    for(i = 0; i < len; i++){
        index = s2[i] - 'a';
        if(p->next!=NULL){
            if(i == len - 1){
                return p->next[index]->num;
            }
        }
        else{
            return 0;
        }
        p = p->next[index];
    }
    return 0;
}

int main()
{
    node *root = new node();
    while(getline(cin, s1) && !s1.empty()){
        Insert(root, s1);
    }
    while(getline(cin, s2)){
        int k = Find(root, s2);
        printf("%d\n", k);
    }
    return 0;
}