HDOJ1251 统计难题 Trie树的基本操作

题目传送门：
http://acm.hdu.edu.cn/showproblem.php?pid=1251

Trie树是一种特殊的多叉树结构，又称字典树或前缀树。特点是占用内存大（可以尝试左儿子右兄弟存储），但是可以很好的节约时间，时间复杂度为O（N），N为查找或插入的字符串长度。

题目中统计含某一前缀的单词数正是Trie树的典型应用。

Trie树的定义：

typedef struct Trie{
//记录含有该前缀的单词数目
    int cnt;              
/*指针域，表示26个小写字母，NULL意味着字典中并没有以该前缀开头的单词*/
    struct Trie *next[26];
} Trie,*pTrie;

基本操作：

创建Trie树：

//返回根节点指针
pTrie Init(void)
{
    pTrie p = (pTrie)malloc(sizeof(Trie));
    p -> cnt = 0;
    for (int i = 0; i < 26; ++i)
    {
        p -> next[i] = NULL;
    }
    return p;
}

添加单词：

void Add(pTrie p, char *s)
{
    int len = strlen(s);
    for (int i = 0; i < len; ++i)
    {
        int id = s[i] - 'a';
        // 从首字母开始扫描，当遇到新的前缀时新增结点
        if (p->next[id] == NULL) {
            pTrie q = (pTrie)malloc(sizeof(Trie));
            q -> cnt = 1;
            int j;
            for (j = 0; j < 26; j ++) {
                q -> next[j] = NULL;
            }
            p -> next[id] = q;
            p = p -> next[id];
        } else { //当前前缀已存在，cnt ++
            p -> next[id] -> cnt ++;
            p = p -> next[id];
        }
    }
    return ;
}

查找：

//树的遍历，返回以s为前缀的单词数目
int Find(pTrie p, char *s)
{
    int len = strlen(s);
    for (int i = 0; i < len; ++i)
    {
        int id = s[i] - 'a';
        if (p -> next[id] == NULL) {
            return 0;
        } else {
            p = p->next[id];
        }
    }
    return p -> cnt;
}

有了这些基本操作，题目也就简单了。

AC代码

// TRIE

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>

using namespace std;

typedef struct Trie{
    int cnt;
    struct Trie *next[26];
} Trie,*pTrie;

pTrie Init(void)
{
    pTrie p = (pTrie)malloc(sizeof(Trie));
    p -> cnt = 0;
    for (int i = 0; i < 26; ++i)
    {
        p -> next[i] = NULL;
    }
    return p;
}

void Add(pTrie p, char *s)
{
    int len = strlen(s);
    for (int i = 0; i < len; ++i)
    {
        int id = s[i] - 'a';
        if (p->next[id] == NULL) {
            pTrie q = (pTrie)malloc(sizeof(Trie));
            q -> cnt = 1;
            int j;
            for (j = 0; j < 26; j ++) {
                q -> next[j] = NULL;
            }
            p -> next[id] = q;
            p = p -> next[id];
        } else {
            p -> next[id] -> cnt ++;
            p = p -> next[id];
        }
    }
    return ;
}

int Find(pTrie p, char *s)
{
    int len = strlen(s);
    for (int i = 0; i < len; ++i)
    {
        int id = s[i] - 'a';
        if (p -> next[id] == NULL) {
            return 0;
        } else {
            p = p->next[id];
        }
    }
    return p -> cnt;
}

int main(int argc, char const *argv[])
{
    pTrie p = Init();
    char s[100];
    while (gets_s(s),s[0] != '\0') {
        Add(p,s);
    }
    while (~scanf("%s",s)) {
        cout << Find(p,s) << endl;
    }

    return 0;
}