统计难题
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 15031 Accepted Submission(s): 6436
Problem Description
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input
banana
band
bee
absolute
acm
ba
b
band
abc
Sample Output
2 3 1 0
思路：字典树
AC代码：
1 #include<stdio.h> 2 #include<string.h> 3 #include<stdlib.h> 4 typedef struct Tree_Node 5 { 6 int cnt; 7 struct Tree_Node *child[30]; 8 }Node; 9 Node *root; 10 char str[15]; 11 void insert() 12 { 13 int i; 14 if(str == NULL) 15 return ; 16 char *p = str; 17 Node *t = root; 18 while(*p != ' ') 19 { 20 if(NULL == t -> child[*p- 'a']) 21 { 22 Node *temp = (Node *)malloc(sizeof(Node)); 23 memset(temp, 0, sizeof(Node)); 24 for(i = 0;i < 30;i ++) 25 { 26 temp -> child[i] = NULL; 27 temp -> cnt = 0; 28 } 29 t -> child[*p - 'a'] = temp; 30 } 31 t = t -> child[*p - 'a']; 32 t -> cnt ++; 33 p ++; 34 } 35 } 36 int search() 37 { 38 if(NULL == root) 39 return ; 40 char *p = str; 41 Node *t = root; 42 while(*p != ' ') 43 { 44 if(NULL != t -> child[*p - 'a']) 45 { 46 t = t -> child[*p - 'a']; 47 p ++; 48 } 49 else 50 return 0; 51 } 52 return t -> cnt; 53 } 54 55 int main(int argc, char const *argv[]) 56 { 57 int i; 58 char c[3] = "abc"; 59 Node TREEROOT; 60 root = &TREEROOT; 61 for(i = 0;i < 30;i ++) 62 { 63 root -> child[i] = NULL; 64 root -> cnt = 0; 65 } 66 // freopen("in.c","r",stdin); 67 while(gets(str) && strcmp(str,"")) 68 { 69 insert(); 70 memset(str, 0, sizeof(str)); 71 } 72 memset(str, 0, sizeof(str)); 73 while(~scanf("%s", str)) 74 { 75 printf("%d ", search()); 76 memset(str, 0, sizeof(str)); 77 } 78 return 0; 79 }