字典树

最新推荐文章于 2020-07-26 08:57:40 发布

原创最新推荐文章于 2020-07-26 08:57:40 发布 · 376 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#字典树 #ACM

字典树专栏收录该内容

1 篇文章

订阅专栏

本文介绍了两种实现字典树的方法：一种是使用数组保存节点信息，另一种是采用动态指针方式。通过示例代码详细展示了如何利用这两种方法进行单词插入与查找操作，并提供了一个完整的字典树与KMP算法的学习课件。

字典树与KMP模板

字典树与KMP PPT课件

1.数组保存

#include <stdio.h>
#include <string.h>
const int maxn=10000;//提前估计好可能会开的节点的个数

int tot;            //节点编号，模拟申请新节点，静态申请
int trie[maxn][26]; //假设每个节点的分支有26个
bool isw[maxn];     //判断该节点是不是单词结尾

void insert(char *s,int rt){
    for(int i=0;s[i];i++){
        int x=s[i]-'a';//假设单词都是小写字母组成
        if(trie[rt][x]==0){//没有，申请新节点
            trie[rt][x]=++tot;
        }
        rt=trie[rt][x];
    }
    isw[rt]=true;
}

bool find(char *s,int rt){
    for(int i=0;s[i];i++){
        int x=s[i]-'a';//假设单词都是小写字母组成
        if(trie[rt][x]==0){
            return false;
        }
        rt=trie[rt][x];
    }
    return isw[rt];
}

char s[22];//单词读入

int main(){
    tot=0;//一开始没有节点

    int rt=++tot;//申请一个根节点
    memset(trie[rt],0,sizeof(trie[rt]));//初始化根节点
    memset(isw,false,sizeof(isw));

    while(scanf("%s",s),s[0]!='#'){//新建字典，以一个'#'结束
        insert(s,rt);
    }
    while(scanf("%s",s),s[0]!='#'){//查单词，以一个'#'结束
        if(find(s,rt))
            printf("%s 在字典中\n",s);
        else
            printf("%s 不在字典中\n",s);
    }
    return 0;
}

2.动态指针

///hdu1251
#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;
}

字典树KMP优先队列完整学习课件打包：点击下载课件压缩包