字典树

字典树又称单词查找树,Trie树,是一种树形结构,是一种哈希树的变种。典型应用是用于统计,排序和保存大量的字符串(但不仅限于字符串),所以经常被搜索引擎系统用于文本词频统计。它的优点是:利用字符串的公共前缀来节约存储空间,最大限度地减少无谓的字符串比较,查询效率比哈希表高。 


字典树与字典很相似,当你要查一个单词是不是在字典树中,首先看单词的第一个字母是不是在字典的第一层,如果不在,说明字典树里没有该单词,如果在就在该字母的孩子节点里找是不是有单词的第二个字母,没有说明没有该单词,有的话用同样的方法继续查找.字典树不仅可以用来储存字母,也可以储存数字等其它数据。

Trie的数据结构定义:

#define MAX 26
typedef struct Trie   
{   
    Trie *next[MAX];   
    int v;   //根据需要变化
}Trie;   
 
Trie *root;
next是表示每层有多少种类的数,如果只是小写字母,则26即可,若改为大小写字母,则是52,若再加上数字,则是62了,这里根据题意来确定。
v可以表示一个字典树到此有多少相同前缀的数目,这里根据需要应当学会自由变化。

Trie的查找(最主要的操作):

(1) 每次从根结点开始一次搜索;
(2) 取得要查找关键词的第一个字母,并根据该字母选择对应的子树并转到该子树继续进行检索;   

(3) 在相应的子树上,取得要查找关键词的第二个字母,并进一步选择对应的子树进行检索。   

(4) 迭代过程……   

(5) 在某个结点处,关键词的所有字母已被取出,则读取附在该结点上的信息,即完成查找。

这里给出生成字典树和查找的模版:

生成字典树:

void createTrie(char *str)
{
    int len = strlen(str);
    Trie *p = root, *q;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'0';
        if(p->next[id] == NULL)
        {
            q = (Trie *)malloc(sizeof(Trie));
            q->v = 1;    //初始v==1
            for(int j=0; j<MAX; ++j)
                q->next[j] = NULL;
            p->next[id] = q;
            p = p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p = p->next[id];
        }
    }
    p->v = -1;   //若为结尾,则将v改成-1表示
}


接下来就是查找过程了:

int findTrie(char *str)
{
    int len = strlen(str);
    Trie *p = root;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'0';
        p = p->next[id];
        if(p == NULL)   //若为空集,表示不存以此为前缀的串
            return 0;
        if(p->v == -1)   //字符集中已有串是此串的前缀
            return -1;
    }
    return -1;   //此串是字符集中某串的前缀
}

对于上述动态字典树,有时会超内存,比如HDU 1671 Phone List,需要释放空间

int dealTrie(Trie* T)
{
    int i;
    if(T==NULL)
        return 0;
    for(i=0;i<MAX;i++)
    {
        if(T->next[i]!=NULL)
            deal(T->next[i]);
    }
    free(T);
    return 0;
}


OJ字典树题集:

1.http://acm.hdu.edu.cn/showproblem.php?pid=1251

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;

typedef struct Trie
{
    Trie *child[26];
    int n;
}Trie;
Trie *root;

void createTrie(char *source)
{
    int len,i,j;
    Trie *current,*newnode;
    len=strlen(source);
    if(len==0) return;
    current=root;
    for(i=0;i<len;i++)
    {
        if(current->child[source[i]-'a']!=0)
        {
            current=current->child[source[i]-'a'];
            current->n=current->n+1;
        }
        else
        {
            newnode=(Trie *)malloc(sizeof(Trie));
            for(j=0;j<26;j++)
                newnode->child[j]=0;
            current->child[source[i]-'a']=newnode;
            current=newnode;
            current->n=1;
        }
    }
}
int findTrie(char *source)
{
    int i,len;
    Trie *current;
    len=strlen(source);
    if(len==0) return 0;
    current=root;
    for(i=0;i<len;i++)
    {
        if(current->child[source[i]-'a']!=0)
            current=current->child[source[i]-'a'];
        else return 0;
    }
    return current->n;
}

int main()
{
    char temp[11];
    int i;
    root=(Trie *)malloc(sizeof(Trie));
    for(i=0;i<26;i++)
        root->child[i]=0;
    while(gets(temp),strcmp(temp,"")!=0)
        createTrie(temp);
    while(scanf("%s",temp)!=EOF)
    {
        i=findTrie(temp);
        printf("%d\n",i);
    }
}

2. http://acm.hdu.edu.cn/showproblem.php?pid=1671

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;

typedef struct Trie
{
    Trie *child[26];
    int n;
}Trie;
Trie *root;

void createTrie(char *source)
{
    int len,i,j;
    Trie *current,*newnode;
    len=strlen(source);
    current=root;
    for(i=0;i<len;i++)
    {
        int id = source[i]-'0';
        if(current->child[id]!=NULL)
        {
            current=current->child[id];
            current->n++;
        }
        else
        {
            newnode=(Trie *)malloc(sizeof(Trie));
            newnode->n = 1;
            for(j=0;j<10;j++)
                newnode->child[j]=0;
            current->child[id]=newnode;
            current=newnode;
        }
    }
    current->n = -1;
}
int findTrie(char *source)
{
    int i,len;
    Trie *current;
    len = strlen(source);
    current = root;
    for(i=0; i < len; i++)
    {
        int id = source[i] - '0';
        current = current->child[id];
        if(current == NULL)
            return 0;
        if(current->n == -1)
            return -1;
    }
    return -1;
}

int deal(Trie* T)
{
    int i;
    if(T==NULL)
        return 0;
    for(i=0;i<10;i++)
    {
        if(T->child[i]!=NULL)
            deal(T->child[i]);
    }
    free(T);
    return 0;
}

int main()
{
    char temp[15];
    int i;
    bool flag;
    int ncase, all;
    scanf("%d", &ncase);
    while(ncase--)
    {
        root=(Trie *)malloc(sizeof(Trie));
        for(i=0;i<10;i++)
            root->child[i]=NULL;
        flag = 0;
        scanf("%d", &all);
        for(int i = 0; i < all; i++)
        {
            scanf("%s", temp);
            if(findTrie(temp) == -1)
            {
                flag = 1;
            }
            if(flag)
                continue;
            createTrie(temp);
        }
        deal(root);
        if(flag)
            printf("NO\n");
        else
            printf("YES\n");
    }
    return 0;
}



评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值