HDU 1251 统计难题（字典树计算前缀数量）

最新推荐文章于 2022-01-10 05:11:12 发布

转载最新推荐文章于 2022-01-10 05:11:12 发布 · 125 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jifahu/p/5449340.html

本文介绍了一种使用字典树（Trie）的数据结构来存储和查找字符串的方法。具体实现了插入和查找功能，其中每个节点记录了以此节点为前缀的字符串数量。通过此方法可以高效地进行字符串匹配。

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

　　字典树应用,每个节点上对应的cnt是以它为前缀的单词的数量

#include<stdio.h>
#include<string.h>
struct trie
{
    int cnt;
    trie *next[26];
};
trie *root=new trie;
void insert(char ch[])
{
    trie *p=root,*newnode;
    for(int i=0; ch[i]!='\0'; i++)
    {
        if(p->next[ch[i]-'a']==0)
        {
            newnode=new trie;
            for(int j=0; j!=26; j++)
            {
                newnode->next[j]=NULL;
            }
            newnode->cnt=1;
            p->next[ch[i]-'a']=newnode;
            p=newnode;
        }
        else
        {
            p=p->next[ch[i]-'a'];
            p->cnt++;
        }
    }
}
int find(char ch[])
{
    trie *p=root;
    for(int i=0; ch[i]!='\0'; i++)
    {
        if(p->next[ch[i]-'a']!=NULL)
            p=p->next[ch[i]-'a'];
        else
            return 0;
    }
    return p->cnt;
}
int main()
{
    char ch[20];
    for(int i=0; i!=26; i++)
    {
        root->next[i]=NULL;
    }
    root->cnt=0;
    while(gets(ch)) //±ØÐëÓÃgets
    {
        if(!strcmp(ch,"")) break;
        insert(ch);
    }
    while(scanf("%s",ch)!=EOF)
    {
        printf("%d\n",find(ch));
    }
    return 0;
}