hdu 1251 Trie入门(统计前缀出现次数)

点击打开链接

#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const int M = 26;
typedef struct Trie{
Trie *next[M]; //节点的子树
int v;//权值

}Trie;
Trie root;
void Insert(char *s)
{
int len=strlen(s);
Trie *p=&root,*q;
for(int i=0;i<len;i++)
{
int id=s[i]-'a';// 子树的编号
if(p->next[id]==NULL)//子树不存在 则create
{
q=(Trie *)malloc(sizeof(Trie));
q->v=1;//前缀出现出现初始化为1
for(int j=0;j<M;j++)
{
q->next[j]=NULL;
}
p->next[id]=q;
}
else
{
p->next[id]->v++; //前缀累加
}
p=p->next[id]; //移动到该子树上 继续插入字符
}

}
int Find(char *s)
{
int len=strlen(s);
Trie *p=&root;
for(int i=0;i<len;i++)
{
int id=s[i]-'a';
p=p->next[id];
if(p==NULL)// 无此前缀
{
return 0;
}
}
return p->v;
}
int main()
{
char s[20];
for(int i=0;i<26;i++)
{
root.next[i]=NULL;
}
while(gets(s)&&s[0]!='\0')
{
Insert(s);
}

while(scanf("%s",s)!=EOF)
{
printf("%d\n",Find(s));
}
return 0;
}