hdu 1251 统计难题（trie 树的简单应用）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251

题意：给你多个字符串，求以某个字符串为前缀的字符串数量。

思路：简单的trie数应用，在trie的数据结构中增加一个存储到当前节点字符串出现的次数，在插入的过程中维护即可。

code：

#include <cstdio>
#include <cstring>
const int KIND = 26;
struct TrieNode
{
    int num;     // 遍历到该结点形成的字符串出现的次数
    TrieNode* next[KIND];
    TrieNode()
    {
        num = 1;
        for (int i = 0; i < KIND; ++i) next[i] = NULL;
    }
};

TrieNode* root = NULL;

void Insert(char* x)
{
    TrieNode* temp = root;
    if (root == NULL)   // 预判
    {
        temp = new TrieNode();
        root = temp;
    }
    int len = strlen(x);
    for (int i = 0; i < len; ++i)
    {
        int curr = x[i] - 'a';
        if (temp->next[curr] != NULL) ++(temp->next[curr]->num);  // 已经存在
        else temp->next[curr] = new TrieNode();  // 不存在
        temp = temp->next[curr];
    }
}

int Search(char* x)
{
    int ret = 0;
    if (root == NULL) return ret;   // 预判
    TrieNode* temp = root;
    int len = strlen(x);
    for (int i = 0; i < len; ++i)
    {
        int curr = x[i] - 'a';
        if (temp->next[curr] == NULL) return 0;
        temp = temp->next[curr];
        ret = temp->num;
    }
    return ret;
}

void Release(TrieNode* root)
{
    if (NULL == root) return;
    for (int i = 0; i < KIND; ++i)
    {
        if (root->next[i] != NULL)
            Release(root->next[i]);
    }
    delete root;
    root = NULL;
}

int main()
{
    char str[10];
    while (gets(str))
    {
        if (str[0] == '')
        {
            while (gets(str))
                printf("%d
", Search(str));
        }
        else Insert(str);
    }
    Release(root);
    return 0;
}