hdu 1251 统计难题(trie)

来源：互联网发布：tsp问题算法编辑：程序博客网时间：2024/04/29 02:55

统计难题

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 20143 Accepted Submission(s): 8807

Problem Description

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

Input

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

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

Output

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

Sample Input

bananabandbeeabsoluteacmbabbandabc

Sample Output

Author

Ignatius.L

题目大意:记录串,并回答前缀为给定串的字符串数目
题目分析:字典树裸题,记录一下当前前缀下的字符串数目即可

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>#include <cctype>#define MAX 500007using namespace std;struct Node{    Node * b[27];    int sum;    Node ( ) : sum (0)    {        memset ( b , 0 , sizeof ( b ) );    }}node[MAX];int cc;int main ( ){    int cc = 0;    char c = getchar ();    Node * root = &node[0];    while ( islower(c) )    {        root->sum++;        while ( islower(c) )         {            if ( !(root->b[c-'a']) )                 root->b[c-'a'] = &node[++cc];            root->b[c-'a']->sum++;            root = root->b[c-'a'];            c = getchar ();        }        root = &node[0];        c = getchar ( );    }    char s[15];    while ( ~scanf ( "%s" , s ) )    {        int len = strlen(s);        root = &node[0];        int i;        for ( i = 0; i < len ; i++ )            if ( root )                root = root->b[s[i]-'a'];            else break;        if ( i == len && root )            printf ( "%d\n" , root->sum );        else puts ( "0" );    }}

0 0