关闭

hdu 1251 统计难题(trie树)

标签: nullstruct
862人阅读 评论(0) 收藏 举报

这两天学了下字典树~

参考连接:

http://www.cnblogs.com/cherish_yimi/archive/2009/10/12/1581666.html

http://www.wutianqi.com/?p=1359

以及hdu的lcy老师的课件。

hdoj的1251求前缀字符串的个数,模版水过,参照dd牛模版



#include<stdio.h>//algo:trie tree
#include<string.h>
#include<stdlib.h>
#define MAX 26
typedef struct Trie{
	Trie *next[MAX];
	int flag;
}Trie;
Trie root;
void createTrie(char *str)
{
	int len=strlen(str);
	Trie *p=&root,*q;
	for(int i=0;i<len;i++)
	{
		int id=str[i]-'a';
		if(p->next[id]==NULL)
		{
			q=(Trie *)malloc(sizeof(Trie));
			q->flag=1;//init
			for(int j=0;j<MAX;j++)
				q->next[j]=NULL;
			p->next[id]=q;
			p=p->next[id];
		}
		else 
		{
			p->next[id]->flag++;
			p=p->next[id];
		}
	}
}
int findTrie(char *str)
{
	int len=strlen(str);
	Trie *p=&root;
	for(int i=0;i<len;i++)
	{
		int id=str[i]-'a';//根据需要减去0,或者‘a’或者'A'
		p=p->next[id];
		if(p==NULL)//若为空集,表示不存在以此为前缀的串
			return 0;
	}
	return p->flag;// 此串是字符集中某串的前缀
}
int dealTrie(Trie* T)//释放空间
{
	int i;
	if(T==NULL)	return 0;
	for(i=0;i<MAX;i++)
	{
		if(T->next[i]!=NULL)
			dealTrie(T->next[i]);
	}
	free(T);
	return 0;
}
int main()
{
	char str[15];
	while(gets(str))
	{
		if(str[0]=='\0') break;
		createTrie(str);
	}
	char aim[15];
	while(~scanf("%s",aim))
	{
		int ans=findTrie(aim);
		printf("%d\n",ans);

	}
	//dealTrie(&root);
	return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:383818次
    • 积分:8273
    • 等级:
    • 排名:第2380名
    • 原创:436篇
    • 转载:24篇
    • 译文:0篇
    • 评论:29条
    文章分类
    最新评论
    even's code life