HDU 1251:统计难题【字典树 & string+map】

统计难题

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 28022    Accepted Submission(s): 11159

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

bananabandbeeabsoluteacmbabbandabc

 

Sample Output

2310

 

AC-code:

map+string:

#include<iostream>#include<cstdio>#include<map>#include<string>#include<cstring>using namespace std;map<string,int>rec;int main(){string goal;char op,str[11];while(1){op=getchar();if(op=='\n'){op=getchar();goal="";}if(op=='\n')break;goal+=op;rec[goal]+=1; }  while(~scanf("%s",str)) { printf("%d\n",rec[str]); } return 0; } 

字典树：

#include<cstdio>#include<cstring>#include<algorithm>#include<iostream>using namespace std;struct trie{trie *next[26];int isword; } *root; void insert(char *tar) { trie *p=root,*newnode; int id; if(strlen(tar)==0) return ; while(*tar) { id=*tar-'a'; if(p->next[id]==0) { newnode=(struct trie *)malloc(sizeof(struct trie)); for(int i=0;i<26;i++) newnode->next[i]=0; p->next[id]=newnode; p=newnode; p->isword=1; } else p=p->next[id],p->isword++; tar++; } } int search(char *tar) { trie *p=root; int id; if(strlen(tar)==0) return 0; while(*tar) { id=*tar-'a'; if(p->next[id]==0) return 0; p=p->next[id]; tar++; } return p->isword; } int main() { char str[11]; root=(struct trie *)malloc(sizeof(struct trie)); int i; for(i=0;i<26;i++) root->next[i]=0; root->isword=2; while(gets(str),strcmp(str,"")!=0) { insert(str); } while(~scanf("%s",str)) { printf("%d\n",search(str)); } }