统计难题
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
banana band bee absolute acm ba b band abc
Sample Output
2 3 1 0
Author
Ignatius.L
题目大意:记录串,并回答前缀为给定串的字符串数目
题目分析:字典树裸题,记录一下当前前缀下的字符串数目即可
题目分析:字典树裸题,记录一下当前前缀下的字符串数目即可
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cctype>
#define MAX 500007
using 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" );
}
}