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;
}