Trie树 Hihocoder 1014 Trie树
题目链接：http://hihocoder.com/problemset/problem/1014
题意：给一些字符串作为字典，再给一些前缀问前缀为这个的有多少个单词。
思路：Trie树模板。
建立的思想不难，用指针指向下一个字符。因为字符长度不定，而且如果是26字母的话就会过长，所以用指针来做最好。那么问题来了。
首先声明一个Trie的时候必须要把他里面标记置为0，把所有指向儿子的Trie指针置为NULL。插入时，使用指针*p指向插入字符串的首地址a，方便接下来开辟空间；用*cur=root来把cur指向root，方便往下遍历。如果当前son为NULL，使用cur->son[*p-’a’] = new Trie进行动态分配。查询同。
源码：
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
int const MAXNODE = 26;
struct Trie
{
int val;
Trie *son[MAXNODE];
Trie(){
val = 0;
for(int i=0; i<MAXNODE; i++)
son[i] = NULL;
}
};
void insert(Trie *root,char *a)
{
if(root == NULL || *a=='\0')
return;
Trie *cur = root;
char *p = a;
for(int i=0; i<strlen(a); i++){
if(cur->son[*p-'a']==NULL)
cur->son[*p-'a'] = new Trie;
cur = cur->son[*p-'a'];
cur->val++;
p++;
}
}
int query(Trie *root,char *a)
{
if(root==NULL || *a=='\0')
return 0;
Trie *cur = root;
char *p = a;
for(int i=0; i<strlen(a); i++){
if(cur->son[*p-'a'] == NULL)
return 0;
cur = cur->son[*p-'a'];
p++;
}
return cur->val;
}
int main()
{
int n,i;
char a[20];
scanf("%d",&n);
Trie *root = new Trie;
while(n--){
scanf("%s",a);
insert(root, a);
}
scanf("%d",&n);
for(int i=0; i<n; i++){
scanf("%s",a);
// getchar();
printf("%d\n",query(root,a));
}
delete root;
return 0;
}