HDU ACM 1251 统计难题(字典树)------------模版

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

字典上的结构为

字典树的模版题

用了指针的指针

注意理解 **Q 的意思

#include <iostream>
#include <string>
using namespace std;
struct Node
{
    int n;
    Node *child[26];
};
void NewNode(Node **Q)//注意这里使用了指针的指针
{
    (*Q) = (Node *)malloc(sizeof(Node));
    (*Q)->n = 1;
    int i;
    for(i=0;i<26;i++)
    {
        (*Q)->child[i] = NULL;
    }
}
void Insert(Node *T,char str[])
{
    Node *Q = T;
    int len = strlen(str);
    int i;
    for(i=0;i<len;i++)
    {
        if(Q->child[str[i] - 'a'] != NULL)
        {
            Q = Q->child[str[i] - 'a'];
            Q->n = Q->n + 1;
        }
        else
        {
            NewNode(&(Q->child[str[i] - 'a']));
            Q = Q->child[str[i] - 'a'];
        }
    }
}

int Search(Node *Q,char str[])
{
    int len = strlen(str);
    int i;
    for(i=0;i<len;i++)
    {
        if(Q->child[str[i] - 'a'] != NULL)
        {
            Q = Q->child[str[i] - 'a'];
        }
        else
        {
            return 0;
        }
    }
    return Q->n;
}
int main()
{
    char str[20];
    Node *T = NULL;
    NewNode(&T);
    while(1)
    {
        gets(str);
        if(str[0] != 0)
        {
            Insert(T,str);
        }
        else
        {
            break;
        }
    }
    while(cin>>str)
    {
        int num = Search(T,str);
        cout<<num<<endl;
    }
    return 0;
}