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;

}