统计难题(HDU1251)

 1 #include <stdio.h>
 2 #include <string.h>
 3 #include <stdlib.h>
 4 #include <malloc.h>
 5 #define MAX 26                  /* 26个字母 */
 6 typedef struct node
 7 {
 8     struct node *child[MAX];    /* 存储下一个字符 */
 9     int n;                      /* 记录当前单词出现的次数 */
10 }node, *Node;
11 Node root;                      /* 字典树的根结点(不存储任何字符) */
12 void insert(char *str)          /* 插入单词 */
13 {
14     int i, index, len;
15     Node t = NULL, newnode = NULL;
16     len = strlen(str);
17     t = root;                   /* 开始时当前的结点为根结点 */
18     for (i = 0; i < len; i++)   /* 逐个字符插入 */
19     {
20         index = str[i] - ‘a‘;   /* 获取此字符的下标 */
21         if (t->child[index] != NULL) /* 字符已在字典树中 */
22         {
23             t = t->child[index];     /* 修改当前的结点位置 */
24             (t->n)++;                /* 当前单词又出现一次, 累加 */
25         }
26         else                         /* 此字符还没出现过, 则新增结点 */
27         {
28             newnode = (Node)calloc(1, sizeof(node)); /* 新增一结点, 并初始化 */
29             t->child[index] = newnode;
30             t = newnode;            /* 修改当前的结点的位置 */
31             t->n = 1;               /* 此新单词出现一次 */
32         }
33     }
34 }
35 int find_word(char *str)/* 在字典树中查找单词 */
36 {
37     int i, index, len;
38     Node t = NULL;
39     len = strlen(str);
40     t = root;                       /* 查找从根结点开始 */
41     for (i = 0; i < len; i++)
42     {
43         index = str[i] - ‘a‘;       /* 获取此字符的下标 */
44         if (t->child[index] != NULL) /* 当前字符存在字典树中 */
45             t = t->child[index];    /* 修改当前结点的位置 */
46         else
47             return 0;               /* 还没比较完就出现不匹配, 字典树中没有此单词 */
48     }
49     return t->n;                    /* 此单词出现的次数 */
50 }
51 void Dele(Node root)                /*释放内存*/
52 {
53     int i;
54     if (NULL == root)
55         return;
56 
57     for (i = 0; i < MAX; i++)
58         if ( root->child[i] != NULL )
59             Dele( root->child[i] );
60     free( root );
61     root = NULL;
62 }
63 int main()
64 {
65     char tmp[11];
66     root = (Node)calloc(1, sizeof(node));
67     while (gets(tmp)&& strlen(tmp))
68         insert( tmp );
69     while (scanf("%s", tmp) != EOF)
70         printf("%d\n", find_word( tmp ));
71     Dele( root );
72     return 0;
73 }