统计难题
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 50524 Accepted Submission(s): 17827
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 | We have carefully selected several similar problems for you: 1075 1247 1671 1298 1800
1 /** 2 算法:字典树 3 4 说明: 5 在字典树模板的基础上要将插入时候的每一步都进行 p->cnt ++ 6 其他与字典树原模板一致 7 **/
核心代码:
1 struct node 2 { 3 node *next[26]; 4 int cnt; 5 node() 6 { 7 cnt = 0; 8 memset(next, 0, sizeof(next)); 9 } 10 }; 11 12 node *root = new node(); 13 14 int my_find(char *s) 15 { 16 node *p = root; 17 int i, k, len = strlen(s); 18 for (i = 0; i < len; ++ i) 19 { 20 k = s[i] - 'a'; 21 if (p->next[k] == NULL) return 0; 22 p = p->next[k]; 23 } 24 return p->cnt; 25 } 26 27 void my_insert(char *s) 28 { 29 node *p = root; 30 int i, k, len = strlen(s); 31 for (i = 0; i < len; ++ i) 32 { 33 k = s[i] - 'a'; 34 if (p->next[k] == NULL) 35 p->next[k] = new node(); 36 p = p->next[k]; 37 p->cnt ++; 38 } 39 return ; 40 }
C/C++代码实现(AC):
#include <iostream> #include <algorithm> #include <cstdio> #include <cstring> using namespace std; int flag = 0, len; char temp[15]; struct node { node *next[26]; int cnt; node() { cnt = 0; memset(next, 0, sizeof(next)); } }; node *root = new node(); int my_find(char *s) { node *p = root; int i, k, len = strlen(s); for (i = 0; i < len; ++ i) { k = s[i] - 'a'; if (p->next[k] == NULL) return 0; p = p->next[k]; } return p->cnt; } void my_insert(char *s) { node *p = root; int i, k, len = strlen(s); for (i = 0; i < len; ++ i) { k = s[i] - 'a'; if (p->next[k] == NULL) p->next[k] = new node(); p = p->next[k]; p->cnt ++; } return ; } int main() { while (gets(temp)) { if (temp[0] == NULL) { flag = 1; continue; } if (!flag) my_insert(temp); else printf("%d ", my_find(temp)); } }