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