HDU 1251 字典树

///trie树 hdu 1251
///套了别人的模板秒过
//题目：
//      给一个单词字典A={str[0], str[1], ...}，再给定一系列的单词B = {s[0], s[1], s[2], ...}, 求字典A中以s[i]为前缀的单词有多少。
//
//解法：
//      Tire tree, 先建字典数，然后再查询每个给定的单词。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;


typedef struct Trie_node
{
    int count;                    // 统计单词前缀出现的次数
    struct Trie_node* next[26];   // 指向各个子树的指针
    bool exist;                   // 标记该结点处是否构成单词
}TrieNode , *Trie;

TrieNode* createTrieNode()
{
    TrieNode* node = (TrieNode *)malloc(sizeof(TrieNode));
    node->count = 0;
    node->exist = false;
    memset(node->next , 0 , sizeof(node->next));    // 初始化为空指针
    return node;
}

void Trie_insert(Trie root, char* word)
{
    Trie node = root;
    char *p = word;
    int id;
    while( *p )
    {
        id = *p - 'a';
        if(node->next[id] == NULL)
        {
            node->next[id] = createTrieNode();
        }
        node = node->next[id];  // 每插入一步，相当于有一个新串经过，指针向下移动
        ++p;
        node->count += 1;      // 这行代码用于统计每个单词前缀出现的次数（也包括统计每个单词出现的次数）
    }
    node->exist = true;        // 单词结束的地方标记此处可以构成一个单词
}

int Trie_search(Trie root, char* word)
{
    Trie node = root;
    char *p = word;
    int id;
    while( *p )
    {
        id = *p - 'a';
        node = node->next[id];
        ++p;
        if(node == NULL)
            return 0;
    }
    return node->count;
}

int main()
{

    Trie root=createTrieNode();
    bool flag=false;
    char str[15];
    while(gets(str))
    {
        if(flag)
            cout<<Trie_search(root,str)<<endl;
        else if(strlen(str)!=0)
             Trie_insert(root,str);
        else
            flag=true;

    }
}