简单的字典树（前缀树）

写这个树，主要是为了完成这道题目。http://hihocoder.com/problemset/problem/1014

代码如下，注释有比较详细的解释

#include <iostream>
#include <string>
#include <typeinfo>
#include <vector>
using namespace std;

/************************************************************************/
/* 树节点是两个数组，一个存储字符对应的下一节点指针
/* 另一个数组存储以该字符结尾的所有前缀次数*/
/************************************************************************/
class TreeNode
{
public:
    TreeNode()
    {
        for (int i=0;i<26;i++)
        {
            nextArray[i]=NULL;
            Count[i]=0;
        }
    }
    TreeNode *nextArray[26];
    int Count[26];
};

class TripTree
{
public:
    TripTree()
    {
        head = NULL;
    }
    void insertStr(const string &str)
    {
        if(NULL == head)
        {
            head = new TreeNode;
        }
        int i=0;
        TreeNode *temp = head;
        const int len = str.length();
        //从头到尾的构造字典树，len-1是为了防止越界
        for(;i<len-1;i++)
        {
            temp->Count[str[i]-'a']++;            
            if (temp->nextArray[str[i]-'a'] == NULL)
            {
                temp->nextArray[str[i]-'a'] = new TreeNode;
            }
            temp = temp->nextArray[str[i]-'a'];
                        
        }
        temp->Count[str[i]-'a']++;//处理最后一个字符，因为是最后一个字符，就没有后继节点，此处不修改对应的下一个指针
    }
    int findPre(const string &str)
    {
        int i=0;
        TreeNode *temp = head;
        const int len = str.length();
        for(;i<len-1;i++)
        {
            if(temp->Count[str[i]-'a'] == 0||temp->nextArray[str[i]-'a'] == NULL)
                return 0;
            temp = temp->nextArray[str[i]-'a'];
        }
        return temp->Count[str[i]-'a'];
    }
    TreeNode *head;

};

int main()
{
    string str;
    int n;
    TripTree obj;
    while (cin >> n)
    {
        while(n--)
        {
            cin >> str;
            obj.insertStr(str);

        }
        cin >> n;
        while(n--)
        {
            cin >> str;
            cout << obj.findPre(str)<<endl;

        }
    }
    return 0;
}