Trie(数组版)

Trie字典树

说到这个应该都不陌生,也许你知道他是什么样的,但用代码实现就有一些生疏!

二叉树

上面就是一个Trie字典树,一般用于实现以下等功能

1.字符串检索

2.词频统计

3.字符串排序

4.前缀匹配

5.最大异或对/最大异或值

Trie原理就是一棵树,用来存储字符串或者二进制,因为其原理与某些问题的解决有天然优势,因此在了解其原理后,我们用代码将其实现!

trie数组表示构建一个trie,num[]可用于表示每个前缀中含有的个数 或者 可用于表示每个字符串结束的位置个数!

const int N=1e6+10;
int trie[N][26];
int num=[N]={0};

接下来是Insert()函数,用于实现trie的插入和删除,插入和删除取决与输入的v;p表示每次插入的下标为p!

通过循环插入字符串,如果trie前缀为出现,则trie[p][n]为pos,不然就是前缀相同,当输入的字符串结束后,num[p]++操作用于储存结束标志!

int pos=1;//存储前缀字符串总个数,相对于传统字典树,每层26个,pos表示一共有多少个字符串,能节约空间 
string s;
void Insert(int v){//v==+1,添加数x;v==-1,删除数x;
	int p=0;
	for(int i=0;i<s.size();i++){
		int n=s[i]-'a';
		if(trie[p][n]==0)//如果没出现就pos++,建立一个新的点 
			trie[p][n]= pos++;
		p=trie[p][n];//寻找这个前缀字符串在数组字典树中的下标 
		num[p]+=v;//统计出现个数 
	}
}

插入操作理解后,Find()查询操作就类似了,通过判断一个新的字符串是否在遍历trie过程中有前缀或是是前缀!

int Find(){//查询字符串是否是Trie中的前缀 
	int p=0;
	for(int i=0;i<s.size();i++){//输入字符从开始到结束一步步找 
		int n=s[i]-'a';
		if(trie[p][n]==0) //如果等于0说明此时的前缀字符串没有记录 
			return 0;
		p=trie[p][n];
	}
	return num[p];//找到了完整的字符串下标对应的个数 
}

最终代码如下:

#include<bits/stdc++.h>//数组版字典树 p198 
/*
1.字符串检索
2.词频统计
3.字符串排序
4.前缀匹配 
*/
using namespace std;
const int N=1e6+10;
int trie[N][26];
int num=[N]={0};
int pos=1;//存储前缀字符串总个数,相对于传统字典树,每层26个,pos表示一共有多少个字符串,能节约空间 
string s;
void Insert(int v){//v==+1,添加数x;v==-1,删除数x;
	int p=0;
	for(int i=0;i<s.size();i++){
		int n=s[i]-'a';
		if(trie[p][n]==0)//如果没出现就pos++,建立一个新的点 
			trie[p][n]= pos++;
		p=trie[p][n];//寻找这个前缀字符串在数组字典树中的下标 
		num[p]+=v;//统计出现个数 
	}
}

int Find(){//查询字符串是否是Trie中的前缀 
	int p=0;
	for(int i=0;i<s.size();i++){//输入字符从开始到结束一步步找 
		int n=s[i]-'a';
		if(trie[p][n]==0) //如果等于0说明此时的前缀字符串没有记录 
			return 0;
		p=trie[p][n];
	}
	return num[p];//找到了完整的字符串下标对应的个数 
}


int main(){
	
	return 0;
}

这是模板的代码,稍加改变可以实现很多功能,最主要还是要理解其中的意思,以后就可以得心应手了!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值