zoukankan      html  css  js  c++  java
  • 字典树(Trie树) C++实现



    说明:

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




    代码如下:

    //插入、删除、查询、遍历四种操作
    //注意:四种操作的函数实现中,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;
    }
    





  • 相关阅读:
    复杂对象类型的WebService高级部分
    linux 免输入密码脚本
    查看端口是否被占用
    shell将脚本输出结果记录到日志文件
    多线程注意点
    apache Tomcat配置SSL(https)步骤
    常用的web安全处理
    SQL 中的 UNION 和UNION ALL 的区别
    数据库和数据仓库区别
    Oracle数据库创建表是有两个约束带有默认索引
  • 原文地址:https://www.cnblogs.com/DOLFAMINGO/p/7538623.html
Copyright © 2011-2022 走看看