HDoj 1251統計難題【辞書ツリー基礎題】

統計上の課題
Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 27878    Accepted Submission(s): 11122
Problem Description
Ignatiusは最近、ある文字列を接頭辞とする単語の数(単語自体も自分の接頭辞)を統計するように要求された.
 
Input
入力データの第1部は1枚の単語表で、1行ごとに1つの単語、単語の長さは10を超えないで、それらは先生がIgnatiusに渡して統計した単語を代表して、1つの空行は単語表の終わりを代表します.第2部は一連の質問で、各行は1つの質問で、各質問はすべて1つの文字列です.
注意:本題はテストデータのセットだけで、ファイルが終わるまで処理します.
 
Output
各質問に対して、その文字列を接頭辞とする単語の数を与える.
 
Sample Input

   
   
   
   

    
    
    
    banana
band
bee
absolute
acm

ba
b
band
abc
   
   
   
   

 
Sample Output

   
   
   
   

    
    
    
    2
3
1
0
   
   
   
   

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#define mem(a, b) memset(a, (b), sizeof(a))
#define Wi(a) while(a--)
#define Si(a) scanf("%d", &a)
#define Pi(a) printf("%d
", (a))
#define INF 0x3f3f3f3f
#include<algorithm>
using namespace std;
const int M = 1000000+10;
char str[100];//      
char ask[100];//        
int ch[M][30];//       
int vis[M];//       
int word[M];//              
int sz;//    
void init()
{
	sz = 1;
	mem(ch[0], 0);
	mem(word, 0);
}
int idx(char x)
{
	return x-'a';
}
void insert(char *s)
{
	int i, j, len = strlen(s);
	int u = 0;//       
	for(i = 0; i < len; i++)
	{
		int c = idx(s[i]);
		if(!ch[u][c])//        
		{
			mem(ch[sz], 0);
			vis[sz] = 0;
			ch[u][c] = sz++;//   +1 
		}
		u = ch[u][c];
		word[u]++;//   +1 
	}
	vis[u] = 1;//       
}
int find(char *s)//     s              
{
	int i, j, len = strlen(s);
	int u = 0;
	for(i = 0; i < len; i++)
	{
		int c = idx(s[i]);
		if(!ch[u][c])	return 0;
		u = ch[u][c];
	}
	return word[u];
}
int main()
{
	init();
	while(gets(str), str[0])
	{
		insert(str);
	}
	while(scanf("%s", ask)!=EOF)
	{
		printf("%d
", find(ask));
	} 
	return 0;
}