hihoCoder 1014 Trie树

hihoCoder刷题开始，希望自己能一直坚持下去。关于Trie树，主要是结构体的树的相关应用，可以用于数据搜索，在实现的时候，还做了一些借鉴= =（很长时间不用C++了，而且当时学的也不好），然后结合自己的理解，写了一下注释。

#include<iostream>
#include<cstring>
#include<vector>
#include<stdio.h>
using namespace std;
const int N=26;//字母种类
char dict[15];//单词长度
int n,m;
struct node
{
	int num;//以当前前缀开头的单词计数
	node *child[N];//当前节点叶
}tree_gen;

int check_dict(char *p)//查询
{
node *node_p=&tree_gen;
int ans=0;
while(*p!='\0')
{
if (node_p->child[*p-'a']!=0)//当前节点，是否存在*p保存的字母
{
node_p=node_p->child[*p-'a'];//指向下一个节点
ans=node_p->num;//赋值
}
else
{
return 0;
}
p++;
}
return ans;//返回查询个数
}
int insert_tree(char *p)
{
node *node_p=&tree_gen;
node_p->num++;//计数+1
while(*p!='\0')
{
if (node_p->child[*p-'a']==0)//当前节点不存在此字母
{
node *new_node=new(node);//添加新节点
for (int i=0;i<N;i++)
	new_node->child[i]=0;//初始化
new_node->num=0;
node_p->child[*p-'a']=new_node;
node_p=new_node;
}
else
	node_p=node_p->child[*p-'a'];
node_p->num++;//计数+1
p++;
}
return 0;
}

int main()
{
tree_gen.num=0;
for(int i=0;i<N;i++) 
{
	tree_gen.child[i]=0;
}
cin>>n;
getchar();
for(int i=0;i<n;i++)
{
scanf("%s",dict);//读取输入字符串
insert_tree(dict);
}	
cin>>m;
getchar();
for(int i=0;i<m;i++)
{
scanf("%s",dict);
cout<<check_dict(dict)<<endl;

}
return 0;
}

原出处:https://www.cnblogs.com/xcw0754/p/4443062.html