hihocoder Trie树

思路：

Trie树（字典树）。

数据量大的时候可以预先开一个全局数组存放树节点，需要的时候就分配，比每次new申请空间省时间。

实现：

#include <iostream>
#include <cstdio>
#include <string.h>
#define MAX 26

using namespace std;

struct trieNode
{
    trieNode * next[MAX];
    int cnt;
}; 

trieNode * createTrie()
{
    trieNode * root = new trieNode();
    root -> cnt = 1;
    for(int i = 0; i < MAX; i++)
    {
        root -> next[i] = NULL;    
    }
    return root;
} 

void insert(trieNode * root,char * str)
{
    trieNode * tmp = root;
    int len = strlen(str);
    for(int i = 0; i < len; i++)
    {
        int index = str[i] - 'a';
        if(tmp -> next[index] != NULL)
        {
            tmp -> next[index] -> cnt ++;
        }
        else
        {
            tmp -> next[index] = createTrie();
        }
        tmp = tmp -> next[index];
    }
}

int search(trieNode * root,char * str)
{
    int len = strlen(str);
    trieNode * tmp = root;
    for(int i = 0; i < len; i++)
    {
        int index = str[i] - 'a';
        if(tmp -> next[index] != NULL)
        {
            tmp = tmp -> next[index]; 
        }
        else
        {
            return 0;
        }
    }
    return tmp -> cnt;
}
int main()
{
    trieNode * root = createTrie();
    char tmp[11];
    int n, m;
    scanf("%d", &n);
    for(int i = 0; i < n; i++)
    {
        scanf("%s", tmp);
        insert(root, tmp);
    }
    scanf("%d", &m);
    for(int i = 0; i < m; i++)
    {
        scanf("%s", tmp);
        printf("%d
", search(root, tmp));
    }
    return 0;
}