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;
}