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



说明:

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




代码如下:

//插入、删除、查询、遍历四种操作
//注意:四种操作的函数实现中,T都是指向上一个结点的指针,以此方便操作。
#include
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<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) <





你可能感兴趣的:(数据结构)