HDU1251统计难题 trie树

Description

Ignatius最近遇到一个难题, 老师交给他很多单词( 只有小写字母组成, 不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量( 单词本身也是自己的前缀).

Input

输入数据的第一部分是一张单词表, 每行一个单词, 单词的长度不超过10, 它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问, 每行一个提问, 每个提问都是一个字符串. 

注意: 本题只有一组测试数据, 处理到文件结束.

Output

对于每个提问, 给出以该字符串为前缀的单词的数量.

题意就是。。。喵的中文题目好吧

trie树的模版题，建一个保存指针的数组。每次存储访问节点次数即可

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<vector>
#include<string>
using namespace std;
typedef long long ll;
#define N 505
const int inf=0x3f3f3f3f;
struct node
{
int num;
node* next[26];
node()
{
num=0;
memset(next,NULL,sizeof(next));
}
};
node *root=new node();
node *pos;
void build(string s)
{
pos=root;
for(int i=0;i<s.size();i++)
{
int index=s[i]-'a';
if(pos->next[index]==NULL)
{
pos->next[index]= new node();
pos=pos->next[index];
pos->num++;
}
else
{
pos=pos->next[index];
pos->num++;
}
}
}
int query(string s)
{
pos=root;
for(int i=0;i<s.size();i++)
{
int index=s[i]-'a';
if(pos->next[index]==NULL) return 0;
pos=pos->next[index];
}
return pos->num;
}
int main()
{
int T;
string s;
while(getline(cin,s)&&s[0]!='\0')
{
//printf("fuck\n");
build(s);
}
while(cin>>s)
{
printf("%d\n",query(s));
}
return 0;
}