【基础数据结构】Trie字典树

萌新好菜

已于 2024-08-10 14:19:15 修改

阅读量57

点赞数

文章标签：算法 c++ 学习

于 2023-08-10 16:24:02 首次发布

本文链接：https://blog.csdn.net/m0_73896521/article/details/132212799

版权

本文介绍了一段C++代码，使用字典树数据结构处理字符串的添加和查找操作，通过nxt数组表示节点连接和isend数组标记结束节点。

摘要由CSDN通过智能技术生成

题目： http://oj.daimayuan.top/course/7/problem/537

思想：字典树添加与查找

代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

const int N = 5e5+5;
const int inf = 0x3f3f3f3f;

int n,m;
int nxt[N][26];
//nxt[i][j]=k,表示编号为i的节点的第j个孩子是编号为k的节点

int cnt;
bool isend[N];
string a;

int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a;
		int len=a.size();
		int now=0;
		for(int j=0;j<len;j++){
			int x=a[j]-'a';
			if(!nxt[now][x]){
				nxt[now][x]=++cnt;
			}
			now=nxt[now][x];
		}
		isend[now]=true;
	}
	cin>>m;
	for(int i=1;i<=m;i++){
		cin>>a;
		int len=a.size();
		int now=0;
		bool ok=true;
		for(int j=0;j<len&&ok;j++){
			int x=a[j]-'a';
			if(!nxt[now][x]){
				ok=false;
			}
			else{
				now=nxt[now][x];
			}
		}
		if(ok){
			if(!isend[now]){
				ok=false;
			}
		}
		cout<<ok<<"\n";
	}
	
	return 0;	

}

练习abc：https://atcoder.jp/contests/abc287/tasks/abc287_e