hdoj_1251統計的な課題

統計上の課題
Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others) Total Submission(s): 12722    Accepted Submission(s): 5378
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<iostream>
#include<cstdio>
using namespace std;
#pragma warning(disable : 4996)
#define MAX 26

typedef struct Node
{
	struct Node *next[MAX];
	int v;
}Node, *Trie;
Trie root;

void createTrie(char *str)
{
	int len, i, j;
	Node *current, *newnode;
	len = strlen(str);
	if(len == 0)
	{
		return;
	}
	current = root;
	for(i = 0; i < len; i++)
	{
		int id = str[i] - 'a';
		if(current->next[id] == NULL)
		{
			newnode = (Trie)malloc(sizeof(Node));
			newnode->v = 1;    //  v==1
			for(j = 0; j < MAX; j++)
			{
				newnode->next[j] = NULL;
			}
			current->next[id] = newnode;
			current = current->next[id];
		}
		else
		{
			current->next[id]->v++;
			current = current->next[id];
		}
	}
	 //current->v = -1;   //    ，  v  -1  (     )
}

int findTrie(char *str)
{
	int i, len;
	Trie current = root;
	len = strlen(str);
	if(len == 0)
	{
		return 0;
	}
	for(i = 0; i < len; i++)
	{
		int id = str[i] - 'a';  //         '0'  'a'，   'A'
		current = current->next[id];
		if(current == NULL)   //    ，           
		{
			return 0;
		}
		/*if(current->v == -1)   //             
		{
			return -1;
		}*/
	}
	//return -1;   //            
	return current->v;
}

void dealTrie(Trie T)
{
	int i;
	if(T == NULL)
	{
		return;
	}
	for(i = 0; i < MAX; i++)
	{
		if(T->next[i] != NULL)
		{
			free(T->next[i]);
		}
	}
	free(T);
}

int main()
{
	freopen("in.txt", "r", stdin);
	char str[15] = {0};
	int i;
	root = (Trie)malloc(sizeof(Node));
	for (i = 0; i < MAX; i++)
	{
		root->next[i] = NULL;
	}
	while (gets(str) && strcmp(str, "") != 0)
	{
		createTrie(str);
	}
	while (gets(str))
	{
		int ans = findTrie(str);
		printf("%d
", ans);
	}
	return 0;
}