字典树(Trie树) C++实现

来源:互联网 发布:淘宝商城三星手机壳 编辑:程序博客网 时间:2024/06/07 02:09



说明:

以下代码是个人按照自己的理解写的,可能有错误或者不太规范的地方,欢迎指出!




代码如下:

//插入、删除、查询、遍历四种操作//注意:四种操作的函数实现中,T都是指向上一个结点的指针,以此方便操作。#include<bits/stdc++.h>using namespace std;typedef long long LL;const double eps = 1e-6;const int INF = 2e9;const LL LNF = 9e18;const int mod = 1e9+7;const int maxn = 100+10;typedef struct node{    bool exist;     //是否存在单词    int cnt;        //作为多少个单词的前缀,不算自己    struct node *next[26];    void init()    {        exist = cnt = 0;        memset(next,0,sizeof(0));    }}Node, *Tree;void insert(Tree &T, char *s, int k){    if(!T)    {        T = new Node;        T->init();    }    if(!s[k])   //单词尾    {        T->exist = true;    //标记单词存在    }    else    {        T->cnt++;   //作为前缀        insert(T->next[s[k]-'a'], s, k+1);    }}char ans[100];  //存储路径void traverse(Tree T, int k)    //字典序遍历,实际上就是先序遍历{    if(T->exist)   //如果该节点标有单词编号,则输出路径    {        ans[k] = 0;        cout<<ans<<endl;    }    for(int i = 0; i<26; i++)    {        if(T->next[i])        {            ans[k] = i+'a';            traverse(T->next[i], k+1);        }    }}int search(Tree T, char *s, int k){    if(!T) return 0;    //结点不存在,则单词不存在    if(!s[k])       //结点存在,还要看单词是否存在    {        return T->exist;    }    else        //继续往下找    {       return search(T->next[s[k]-'a'], s, k+1);    }}bool del(Tree &T, char *s, int k){    if(!T) return 0;    if(!s[k])    {        if(T->cnt)  //如果作为单词的前缀,则只标记此单词不存在        {            T->exist = 0;        }        else    //如果不作为其他单词的前缀,则直接删除        {            delete T;            T = NULL;        }        return 1;    }    else    {        if(del(T->next[s[k]-'a'], s, k+1))        {            T->cnt--;            if(!T->exist && !T->cnt)    //既不作为单词的前缀,又不作为单词,则直接删除            {                delete T;                T = NULL;            }            return 1;        }        return 0;    }}int main(){    Tree T;    T = new Node;    T->init();    int n;    cin>>n;    char s[100];    for(int i = 1; i<=n; i++)    {        cin>>s;        insert(T,s,0);    }    traverse(T,0);    cin>>s;    del(T,s,0);    traverse(T,0);    cin>>s;    cout<< search(T,s,0) <<endl;}





原创粉丝点击