HDU1251 统计难题（字典树|map

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

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

注意:本题只有一组测试数据,处理到文件结束.
Output对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

2
3
1
0

先贴map做法

 #include<bits/stdc++.h>
 using namespace std;
 map<string,int>mp;
 string s;
 int main()
 {
     while(getline(cin,s)&&s!=""){
         int l =s.size();
         string str;
         for(int i = ;i < l;++i){
             str += s[i];
             mp[str]++;
         }
     }
     while(cin>>s){
         cout<<mp[s]<<endl;
     }
     return ;
 }

 #include<iostream>
 #include<stdio.h>
 #include<string.h>
 char s[];int ans= ;
 struct node{
     int count;
     node *next[];
     node(){for(int i=;i<;i++) next[i]=nullptr;count=;}
 };
 node *p,root;
 void build_tree(){
     int l = strlen(s);
     node *p=&root;
     for(int i = ;i < l;++i){
         if(p->next[s[i]-'a']==NULL){
             p->next[s[i]-'a'] = new node;
         }
         p=p->next[s[i]-'a'];
         p->count++;
     }
 }

 int search(){
     int l = strlen(s);
     node *p=&root;
     for(int i=;i <l;++i){
         if(!p->next[s[i]-'a']) return ;
         p=p->next[s[i]-'a'];
     }
     return p->count;
 }
 int main()
 {
     while(gets(s)){
         if(strlen(s)==) break;
         build_tree();
     }
     while(gets(s)){
         printf("%d\n",search());
     }
     return ;
 }