1123: 统计难题 （字典树）

1123: 统计难题

时间限制: 1 Sec  内存限制: 128 MB
提交: 4  解决: 4
[提交][状态][讨论版]

题目描写叙述

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

输入

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

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

输出

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

例子输入

banana
band
bee
absolute
acm

ba
b
band
abc

例子输出

2
3
1
0

字典树的基础题，已经做了好几个字典树的题，思路也比較清晰，先建立字典树。然后再统计前面的前缀数目。这个题就是输入的格式要注意一下，其它的都比較简单，属于字典的入门题吧。直接用自己曾经的模板；

以下是ac的代码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
typedef struct node //节点的结构体
{
    int count;//计数
    struct node * next[26];
}node;
node * newnode() //创建新节点
{
    node *q;
    q=(node*)malloc(sizeof(node));
    q->count=1;
    for(int i=0;i<26;i++)
        q->next[i]=NULL;
    return q;
}
void build(node *T,char *s) //建立字典树
{
    node *p;
    p=T;
    int len,k;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        k=s[i]-'a';
        if(p->next[k]==NULL)
        {
            p->next[k]=newnode();
            p=p->next[k];
        }
        else
        {
            p=p->next[k];
            p->count++;
        }
    }
}
int search(node *T,char *c) //查询字典树
{
    node *q;
    int sum=0,len,k;
    q=T;
    len=strlen(c);
    for(int i=0;i<len;i++)
    {
        k=c[i]-'a';
        if(q->next[k]!=NULL)
          q=q->next[k];
        else
          return sum=0;
    }
    sum=q->count;
    return sum;
}
int main()
{
    char s[12];
    char c[12];
    node *T;
    T=(node *)malloc(sizeof(node));
    T->count=0;
    for(int i=0;i<26;i++)
        T->next[i]=NULL;
    memset(s,0,sizeof(s));
    memset(c,0,sizeof(c));
    while(gets(s))  //控制输入这里要注意
    {
       if(strcmp(s,"")==0)
        break;
       build(T,s);
    }
    while(scanf("%s",c)!=EOF)
    {
        printf("%d\n",search(T,c));
    }
    return 0;
}