统计难题
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 44119 Accepted Submission(s): 15817
Problem Description
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input
banana band bee absolute acm ba b band abc
Sample Output
2 3 1 0
Author
Ignatius.L
Recommend
Ignatius.L
题目大意:字面意思。。
解题思路:Trie模板题,白书上的建树模板,自己模仿写一个查询功能就行了。。。(不多说了,,去学AC自动机了。现在还没会- -)
/* 输入数据的第一部分是一张单词表. 每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词. 一个空行代表单词表的结束. 第二部分是一连串的提问, 每行一个提问,每个提问都是一个字符串. 对于每个提问,给出以该字符串为前缀的单词的数量. */ #include <cstdio> #include <cstring> using namespace std; const int maxn=1e6+10; int ch[maxn][26],val[maxn],sz; void Init() { memset(ch[0],0,sizeof(ch[0])); memset(val,0,sizeof(val)); sz = 1; } void Insert(char *s) { int n = strlen(s); int u=0; for(int i=0;i<n;i++) { int c = s[i] - 'a'; if(!ch[u][c]) { memset(ch[sz],0,sizeof(ch[sz])); val[sz] = 0; ch[u][c] = sz++; } u = ch[u][c]; val[u]++;//记录根到u这条路u出现了几次,方便直接判断每次询问字符串出现的次数 } } int query(char *s) { int n = strlen(s); int u = 0; if(n>20) return 0; int c; for(int i=0;i<n;i++) { c = s[i] - 'a'; if(!ch[u][c]) return 0; if(i+1==n) break; u = ch[u][c]; } return val[ch[u][c]] ; } int main() { int ans; char s[100]; Init(); while( gets(s)&&s[0]!='\0') { Insert(s); } while(scanf("%s",s)!=EOF) { ans = query(s); printf("%d\n",ans); } }