ACM_模板_字典树

字典树Trie是一门比较简单也比较好懂的算法,因为字典本身是生活中使用率较高的工具,字典的工作原理大家也都懂,其本质也就是前缀查询,对前缀查询的方法,大家很容易可以想到数据结构中的链表结构,用指针去实现一个连接作用。在这里,小编先讲解一下数的指针建立。

首先因为我们字母只有26个,每个节点的指向可以加一个26节点的指针(不考虑大小写),在每输入一个字符串的时候就建立节点。

小编先给出指针实现的字典树模板:

#include <iostream>//指针实现
#include <cstdio>
#include <cstring>
using namespace std;
char str[11];
struct node
{
	node *next[26];
	int cnt;
	node()
	{//构造函数,创建结点时自动执行
		cnt=0;
		for(int i=0; i<26; i++)
			next[i] = NULL;//将该节点的下面的26个节点初始化为空
	}
};
void insert(node *p,char *str)
{
	for(int i=0; str[i]; i++)
	{
		int t=str[i]-'a';
		if(p->next[t] == NULL)
			p->next[t] = new node;//如果节点不存在,那么先new一个
		p = p->next[t];
		p->cnt++;
	}
}
int find(node *p,char *str)
{
	for(int i=0; str[i]; i++)
	{
		int t=str[i]-'a';
		p = p->next[t];
		if(!p) return 0;
	}
	return p->cnt;
}
int main()
{
	node *root=new node();
	while(gets(str)&&strlen(str))
		insert(root,str);
	while(gets(str))
	{
		int ans=find(root,str);
		printf("%d\n",ans);
	}
	return 0;
}

指针由于使用时,速度会很慢,所以我们在使用的时候有时会很可能超时(TLE),所以我们可以云耀到数组去模拟指针地址。

小编这里额给出小编的数组模拟字典树Trie模板。

#include <cstdio>//数组模拟实现 
#include <cstring>
#define N 5000000
int tree[N][26],cnt[N];
char str[20];
int tot;
void insertTrie(char *str)
{
	int len=strlen(str);
	int now=0;
	for(int i=0; i<len; i++)
	{
		int w=str[i]-'a';
		if(tree[now][w] == 0) tree[now][w] = ++tot;
		now = tree[now][w];
		cnt[now]++;
	}
}
int searchTrie(char *str)
{
	int len=strlen(str);
	int now=0;
	for(int i=0; i<len; i++)
	{
		int w=str[i]-'a';
		if(tree[now][w] == 0) return 0;
		now = tree[now][w];
	}
	return cnt[now];
}
int main()
{
	while(gets(str)&&strcmp(str,"")!=0)
		insertTrie(str);
	while(gets(str))
		printf("%d\n",searchTrie(str));
	return 0;
}




  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值