统计难题 hdu1251

   字典树的纯模板题。初学字典树，纪念一下下

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct node
{
	int count;//记录当前单词出现的次数
	node *next[26];
};
node *root;//根节点,不存储信息

node * build_node()//新建节点
{
	node *p=new node;
	for(int i=0;i<26;i++)
		p->next[i]=NULL;
	p->count=1;
	return p;
}

void build_tree(char *s)//建字典树
{
	int len=strlen(s);
	node *p=root;
	for(int i=0;i<len;i++)
	{
		if(p->next[s[i]-'a']==NULL)//该字母未出现过，所以新建一个节点
			p->next[s[i]-'a']=build_node();
		else
			p->next[s[i]-'a']->count++;
		p=p->next[s[i]-'a'];
	}
}

int query(char *s)
{
	int len=strlen(s);
	node *p=root;
	for(int i=0;i<len;i++)
	{
		if(p->next[s[i]-'a']==NULL)
			return 0;
		p=p->next[s[i]-'a'];
	}
	return p->count;
}

int main()
{
	char s[15];
	root=build_node();
	while(gets(s))
	{
		if(s[0]=='\0')
			break;
		build_tree(s);
	}
	while(gets(s))
	{
		printf("%d\n",query(s));
	}
	return 0;
}