HDU 1251 统计难题 【动态存储+字典树（模板）】

统计难题

Description
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

2
3
1

0

题意：给你一个单词表，再随便给你一个串以这个串为前缀的单词有多少少；

思路：建一颗字典数，有相同前缀的合并，每个节点中有前缀个数，查询时从上向下查询就行了；

失误：错了好几次，不是超内存就是非法访问，最后才知道用动态内存写，这一题要掌握动态内存（题目中没有说数据多少，大小时用）还有memset对结构体初始化0的条件。

代码如下：

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

typedef struct tree{
	int num;
	struct tree *son[26];
}Node;

Node *root;         

void initnode(Node *p) {
	int i=0;
	for(i=0;i<26;++i) p->son[i]=NULL;//结构体中有数组 不能用memset()
	p->num=0; 
}
void insert(char *str)//建树 
{
    int i=0; Node *p1=root,*p2; 
	while(str[i]!='\0')
	{
	    int pos=str[i]-'a';
	    if(p1->son[pos]==NULL)
	    {
	    	p2=new Node;
	        initnode(p2);
	        p1->son[pos]=p2;//没赋值 非法访问内存 
		}
		p1=p1->son[pos];
		++p1->num;
		++i;
	 } 	
}

int find(char str[])
{
	int i=0,cnt=0; Node *p=root; 
    while(str[i]!='\0')
    {
    	int pos=str[i]-'a';
    	if(p->son[pos]==NULL) return 0;
    	p=p->son[pos];
    	cnt=p->num;
    	++i;//缺省输出都是0 
	 } 
 	return cnt;
}

int main()
{
	char str[22]; root=new Node;//缺省非法访问内存 
	 initnode(root);
	while(gets(str),strcmp(str,""))
	{
		insert(str);
	}
	while(~scanf("%s",str))
	{
		printf("%d\n",find(str));
	}
	return 0;
}