HDOJ統計難題1251【辞書ツリー】

統計上の課題
Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others) Total Submission(s): 22464    Accepted Submission(s): 9477
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
   
   
   
   

 
Author
Ignatius.L
 
Recommend
Ignatius.L   |   We have carefully selected several similar problems for you:   1247  1671  1298  2846  1711 
 
テンプレートの問題...

#include<stdio.h>
#include<string.h>
#define MAX 1000000+10
using namespace std;
int c[MAX][27];
int v[MAX];
int w[MAX];//         
char s[15];
int tot;
int init()
{
	tot=1;
	memset(c[0],0,sizeof(c[0]));
	memset(w,0,sizeof(w));
} 
int insert(char *s)
{
	int i,temp;
	int l=strlen(s);
	int u=0;
	for(i=0;i<l;i++)
	{
		temp=s[i]-'a';
		if(!c[u][temp])
		{
			memset(c[tot],0,sizeof(c[tot]));
			v[tot]=0;
			c[u][temp]=tot++;
		}
		u=c[u][temp];
		w[u]++;
	}
	v[tot]=1;
}
int search(char *s)
{
	int u=0;
	int l=strlen(s);
	int i,temp;
	for(i=0;i<l;i++)
	{
		temp=s[i]-'a';
		if(!c[u][temp])return 0;
		else u=c[u][temp];
	}
	return w[u];
}
int main()
{
	init();
	
	while(gets(s)&&s[0])
	{
		insert(s);
	}
	while(scanf("%s",s)!=EOF)
	{
		printf("%d
",search(s));
	}
}