题目连接:http://hihocoder.com/problemset/problem/1014
题意:对于每一个给出的字符串,在给出的词典里面找到以这个字符串开头的所有单词的数目。
解析:一个字典树基本题目,根据单词表建字典树,有以下几点:
对于一个节点,从根节点到它本身所经过的单词看做一个前缀,那么所有以此前缀为前缀的单词都在以这个节点为根的子树里.
不妨假设以T为根的子树中标记节点(标记节点即某个单词的最后一个字母代表的节点)的个数为L[T].
即每个节点T用一个整形变量L[T]存储表示以当前字符串(当前字符串即从根节点到节点T的字符串)为前缀的单词的数量,L[T]即为插入单词的时候经过了这个节点T多少次,经过了多少次说明有多少个单词以当前字符串为前缀了.
开始置所有L[T]=0,然后每次添加一个新的单词的时候,都将它经过的所有结点的L[T]全部+1,这样我构建完这棵Trie树的时候,我也就能够同时统计到所有L[T]了.
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
char str[15];
struct Tire{
int num;
Tire* next[26];
Tire()
{
num=0;
for(int i=0;i<26;i++)
next[i]=NULL;
}
}root;
void Insert()
{
Tire* p=&root;
for(int i=0;str[i];i++)
{
int t=str[i]-'a';
if(p->next[t]==NULL)
p->next[t]=new Tire;
p=p->next[t];
p->num++;
}
}
int Find()
{
Tire* p=&root;
for(int i=0;str[i];i++)
{
int t=str[i]-'a';
if(p->next[t]==NULL)
return 0;
p=p->next[t];
}
return p->num;
}
int main()
{
int i,n,m;
scanf("%d",&n);
for(i=0;i<n;i++)
{
scanf("%s",str);
Insert();
}
scanf("%d",&m);
for(i=0;i<m;i++)
{
scanf("%s",str);
printf("%d\n",Find());
}
return 0;
}