学习： ACM-字符串-字典树模板HDU-1251-统计难题

最新推荐文章于 2022-05-06 19:47:57 发布

转载最新推荐文章于 2022-05-06 19:47:57 发布 · 271 阅读

数据结构专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用Trie树的数据结构来实现字符串匹配的方法。通过插入和查询操作演示了如何利用Trie树高效地处理字符串集合。此外，还提供了一个使用数组实现的Trie树示例。

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

#include <cstdio>
#include <cstdlib>
#include <cstring>
 
const int MAXN = 26;
struct Trie
{
    // 代表当前节点可以延伸出的边，数量可变
    Trie *Next[MAXN];
    // 标记当前节点是否是保存信息的结尾，也可以代表前缀个数
    int Flag;
    Trie()
    {
        // 初始化以该信息为前缀的信息个数
        Flag = 1;
        memset(Next, NULL, sizeof(Next));
    }
}*root;
 
void Insert(char *str)
{
    int len = strlen(str);
    Trie *p = root, *q;
    // 将str的每一个字符插入trie树
    for(int i=0; i<len; ++i)
    {
        int id = str[i] - 'a';
        // 如果没有边，则新建一个trie节点，产生一条边，用以代表该字符
        if(p->Next[id] == NULL)
        {
            q = new Trie();
            p->Next[id] = q;
            p = p->Next[id];
        }
        // 如果存在边，则沿着该边走
        else
        {
            p = p->Next[id];
            // 累加以该信息为前缀的信息个数
            ++(p->Flag);
        }
    }
}
 
int Query(char *str)
{
    int len = strlen(str);
    Trie *p = root;
    // 在trie树上顺序搜索str的每一个字符
    for(int i=0; i<len; ++i)
    {
        int id = str[i] - 'a';
        p = p->Next[id];
        // 若为空集，表示不存以此为前缀的信息
        if(p == NULL) return 0;
    }
    // 返回以该信息为前缀的信息个数
    return p->Flag;
}
 
void Free(Trie* T)
{
    if(T == NULL) return;
    // 释放trie树的每一个节点占用的内存
    for(int i=0; i<MAXN; ++i)
    {
        if(T->Next[i]) Free(T->Next[i]);
    }
    delete(T);
}
 
int main()
{//freopen("sample.txt", "r", stdin);
    char str[15];
    // 创建trie树的根节点
    root = new Trie();
    while(gets(str) && str[0]!='\0')
    {
        Insert(str);
    }
    while(~scanf("%s", str))
    {
        printf("%d\n", Query(str));
    }
    Free(root);
    return 0;
}

补充一个删除操作：

删除操作
递归释放字典树的每一个节点占用的空间即可。

void Free(Trie* T)
{
    if(T == NULL) return;
    // 释放trie树的每一个节点占用的内存
    for(int i=0; i<MAXN; ++i)
    {
        if(T->Next[i]) Free(T->Next[i]);
    }
    delete(T);
}

数组的形式:

#include <iostream>
#include <string.h>
#include <cstdio>
#include <bits/stdc++.h>
using namespace std;
int tree[1000010][26];
int num[1000010];
int t = 1;

void Insert(char a[])
{
    int c=0;
    for (int i=0; a[i]; i++)
    {
        int n = a[i]-'a';
        if (tree[c][n]==0) //改串没有存入字典树
        {
            tree[c][n] = t++;
        }
        c = tree[c][n];
        num[c]++;
    }
}

int Find(char a[])
{
    int c=0;
    for (int i=0; a[i]; i++)
    {
        int n = a[i]-'a';
        if (tree[c][n]==0)
        {
            return 0;
        }
         c = tree[c][n];
    }
    return num[c];
}
int main()
{
    char a[11];
    while(gets(a))
    {
        if (a[0]==NULL) 
        {
            break;
        }
        Insert(a);
    }
    while (gets(a))
    {
        printf ("%d\n",Find(a));
    }
    return 0;
}