5.2.2 统计难题

统计难题

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)

Total Submission(s): 207 Accepted Submission(s): 126

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

 

Sample Output

2
3
1
0

思路：Trie或者map，学了两天才明白了trie，其实trie本身很好懂，我的主要时间是花在学习指针和链表上了。

每读入一个单词，在插入时就顺便把前缀的个数加上，这道题就解决了。

另：推荐网站。学习链表和指针：http://see.xidian.edu.cn/cpp/biancheng/view/160.html

trie的话 http://zh.wikipedia.org/wiki/Trie http://dongxicheng.org/structure/trietree/ 的介绍不错

http://blog.163.com/mageng11@126/blog/static/14080837420119544046207/  也很不错

290613

2013-02-17 11:31:29

Accepted

5.2.2

140MS

43868K

1150B

G++

cssystem

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std;

struct node
{
    int cnt;
    node *ch[26];
    node ()
    {
        cnt=0;
        memset(ch,0,sizeof(ch));
    }
};

int l,x;
char s[30];

node *root=new node;
node *cur,*cur1,*newnode;

void close()
{
    exit(0);
}

void insert()
{
    cur=root;
    l=strlen(s);
    for (int i=0;i<l;i++)
    {
        if (s[i]<97 || s[i]>122) break;
        x=s[i]-'a';
        if (cur->ch[x]!=NULL)
        {
            cur=cur->ch[x];
            ++(cur->cnt);
        }
        else
        {
            newnode=new node;
            ++(newnode->cnt);
            cur->ch[x]=newnode;
            cur=newnode;
        }
    }
}

int find()
{
    cur=root;
    l=strlen(s);
    for (int i=0;i<l;i++)
    {
        if (s[i]<97 || s[i]>122) break;
        x=s[i]-'a';
        if (cur->ch[x]==NULL)
            return 0;
        else cur=cur->ch[x];
    }
    return cur->cnt;
}

void init ()
{
   while (fgets(s,30,stdin))
    {
        if (strlen(s)<=1 && (s[0]<97 || s[0]>124) )
            break;
        insert();
        memset(s,'\0',sizeof(s));
    }
    while (fgets(s,30,stdin))
    {
        printf("%d\n",find());
    }
}

int main ()
{
    init();
    close();
    return 0;
}