Hdu-1251 统计难题

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251

解题思路：

字典树的典型应用——插入，查询。

1.插入时候，不是字母结束时将head->count++，而是遇到字母：

没出现过：初始化1，表明此次出现

出现过：count++

这样，head->count记录的就是从根结点到head结点的前缀出现的次数。

2.查询

返回单词结尾字母的count就是前缀出现的次数。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

struct Dictree //结点内容
{
	int count;
	Dictree *tire[26];
}*a; //a为根结点，不存信息，是字典树的入口

void init() //初始化，不能忘。。。。
{
	a = new Dictree;
	for(int i = 0; i < 26; ++i)
		a->tire[i] = NULL;
}

void insert(char str[]) //插入
{
	int len, cur;
	Dictree *head = a; //每次从根结点（a）开始查询
	len = strlen(str);
	for(int i = 0; i < len; ++i)
	{
		cur = (int)(str[i] - 97); //对应字母和下标匹配
		if(head->tire[cur] == NULL) //字母不存在
		{
			head->tire[cur] = new Dictree;
			head = head->tire[cur]; //和下一句顺序不能反
			head->count = 1;
			for(int j = 0; j < 26; ++j)
				head->tire[j] = NULL;
		}
		else //存在
		{
			head = head->tire[cur];
			head->count++;
		}
	}
}

int search(char str[]) //查询
{
	int len, cur;
	Dictree *head = a;
	len = strlen(str);
	for(int i = 0; i < len; ++i)
	{
		cur = (int)(str[i] - 97);
		if(head->tire[cur] == NULL) //无查询结果
			return 0;
		else
			head = head->tire[cur];
	}
	return head->count; //前缀出现次数
}

int main()
{
	char str[11], que[11];
	int temp, query, count;
	init();
	while(gets(str) && strcmp(str, "") != 0) //输入方法不会~~~查了下
		insert(str);
	while(scanf("%s", que) != EOF)
	{
		count = search(que);
		printf("%d\n", count);
	}
	return 0;
}