字典树基础

/*
功能描述:实现trie树的创建、插入、查询
说明:
     对统计对象,要求符合正则"[a-z]*"的格式的单词
     若考虑大写,标点和空白字符(空格.TAB.回车换行符),
     可修改next数组大小,最多255可包含所有字符
*/
#include<stdio.h>
#include <stdlib.h>
#include<string.h>

#define MAX_CHILD 26 //此函数只考虑26个英文字母的情况


typedef struct Tree
{
    int count;         //用来标记该节点是个可以形成一个单词，如果count！=0，则从根节点到该节点的路径可以形成一个单词
    struct Tree *child[MAX_CHILD];
} Node,*Trie_node;
/*
child是表示每层有多少种类的数，如果只是小写字母，则26即可，若改为大小写字母，则是52，若再加上数字，
     则是62了，这里根据题意来确定。
count可以表示一个字典树到此有多少相同前缀的数目，这里根据需要应当学会自由变化。
*/

Node* CreateTrie()                             //创建trie节点树
{
    Node *node=(Node*)malloc(sizeof(Node));
    memset(node,0,sizeof(Node));
    return node;
}

void insert_node(Trie_node root,char *str)      //trie树插入结点
{
    if(root ==NULL || *str==‘\0‘)
        return;

    Node *t=root;
    while(*str!=‘\0‘)
    {
        if(t->child[*str-‘a‘]==NULL)
        {
            Node *tmp=CreateTrie();
            t->child[*str-‘a‘]=tmp;
        }
        t=t->child[*str-‘a‘];
        str++;
    }
    t->count++;
}


void search_str(Trie_node root,char *str)             //查找串是否在该trie树中
{
    if(NULL==root || *str==‘\0‘)
    {
        printf("trie is empty or str is null\n");
        return;
    }

    Node *t=root;
    while(*str!=‘\0‘)
    {
        if(t->child[*str-‘a‘]!=NULL)
        {
            t=t->child[*str-‘a‘];
            str++;
        }
        else
            break;
    }
    if(*str==‘\0‘)
    {
        if(t->count==0)
            printf("该字符串不在trie树中，但该串是某个单词的前缀\n");
        else
            printf("该字符串在该trie树中\n");
    }
    else
        printf("该字符串不在trie树中\n");
}

void del(Trie_node root)      //释放整个字典树占的堆空间
{
    int i;
    for(i=0; i<MAX_CHILD; i++)
    {
        if(root->child[i]!=NULL)
            del(root->child[i]);
    }
    free(root);
}


int main()
{
    int i,n;
    char str[20];
    printf("请输入要创建的trie树的大小：");
    scanf("%d", &n);
    Trie_node root=NULL;
    root=CreateTrie();
    if(root==NULL)
        printf("创建trie树失败\n");
    for(i=0; i<n; i++)
    {
        scanf("%s",str);
        insert_node(root,str);
    }
    printf("trie树已建立完成\n");
    printf("请输入要查询的字符串:\n");
    while(scanf("%s",str)!=NULL)
    {
        search_str(root,str);
    }
    return 0;
 }

#!/usr/bin/env python3

class Trie:
    root = dict()

    def insert(self, string):
        index, node = self.findLastNode(string)
        for char in string[index:]:
            new_node = dict()
            node[char] = new_node
            node = new_node

    def find(self, string):
        index, node = self.findLastNode(string)
        return (index == len(string))

    def findLastNode(self, string):
        ‘‘‘
        @param string: string to be searched
        @return: (index, node).
            index: int. first char(string[index]) of string not found in Trie tree. Otherwise, the length of string
            node: dict. node doesn‘t have string[index].
        ‘‘‘
        node = self.root
        index = 0
        while index < len(string):
            char = string[index]
            if char in node:
                node = node[char]
            else:
                break
            index += 1
        return (index, node)

    def printTree(self, node, layer):
        if len(node) == 0:
            return ‘\n‘

        rtns = []
        items = sorted(node.items(), key=lambda x:x[0])
        rtns.append(items[0][0])
        rtns.append(self.printTree(items[0][1], layer+1))

        for item in items[1:]:
            rtns.append(‘.‘ * layer)
            rtns.append(item[0])
            rtns.append(self.printTree(item[1], layer+1))

        return ‘‘.join(rtns)

    def __str__(self):
        return self.printTree(self.root, 0)

if __name__ == ‘__main__‘:
    tree = Trie()
    while True:
        src = input()
        if src == ‘‘:
            break
        else:
            tree.insert(src)
        print(tree)