力扣hot100 实现Trie(前缀树) 字典树 一题双解

Problem: 208. 实现 Trie (前缀树)

文章目录

  • 思路
  • 复杂度
  • [💝 TrieNode版](#💝 TrieNode版)
  • [💝 二维数组版](#💝 二维数组版)

思路

👩‍🏫 宫水三叶

复杂度

💝 TrieNode版

Java 复制代码
public class Trie
{
	class TrieNode
	{
		boolean end;//标记是否有以当前节点为结尾的字符串
		TrieNode[] ns = new TrieNode[26];
	}

	TrieNode root;

	public Trie()
	{
		root = new TrieNode();
	}

	public void insert(String word)
	{
		TrieNode p = root;
		for (int i = 0; i < word.length(); i++)
		{
			int u = word.charAt(i) - 'a';
//如果正在遍历的该字母在上一个节点的数组坐标中没有记录,就新建一个字母节点在字典树中
			if (p.ns[u] == null)//不存在此节点
				p.ns[u] = new TrieNode();//那就新建节点
			p = p.ns[u];//每一次生成字母都移动指针到下一个字母节点
		}
		p.end = true;// 以当前节点为结尾字符串的个数
	}

	public boolean search(String word)
	{
		TrieNode p = root;
		for (int i = 0; i < word.length(); i++)//枚举所有字符
		{
			int u = word.charAt(i) - 'a';
			if (p.ns[u] == null)//只要有一个字符不存在就 false
				return false;
			p = p.ns[u];
		}
		//最后还要看一下是否以当前字符串结尾
		//例:abcde 路径存在,查 abc,abc存在,但是没有以 c 为结尾的字符串,返回 false
		return p.end;
	}

	public boolean startsWith(String prefix)
	{
		TrieNode p = root;
		for (int i = 0; i < prefix.length(); i++)
		{
			int u = prefix.charAt(i) - 'a';
			if (p.ns[u] == null)
				return false;
			p = p.ns[u];
		}
		//所有字符存在即可,前缀即存在
		return true;
	}
}

/**
 * Your Trie object will be instantiated and called as such:
 * Trie obj = new Trie();
 * obj.insert(word);
 * boolean param_2 = obj.search(word);
 * boolean param_3 = obj.startsWith(prefix);
 */

💝 二维数组版

复制代码
//数组版
public class Trie
{
	int N = 100010;
	int[][] trie;
	int[] cnt;
	int idx = 1;

	public Trie()
	{
//		int[i][j] 表示 第 i 行中的 ('a'+j) 是否存在(默认为 0 表示不存在)
		trie = new int[N][26];
		cnt = new int[N];// cnt[i] 记录 格子 i 被标记为结尾的次数
		idx = 1;// 给用到的格子进行编号(从 0 开始避开默认状态)
	}

	public void insert(String word)
	{
		int p = 0;
		for (int i = 0; i < word.length(); i++)
		{
			int u = word.charAt(i) - 'a';
			if (trie[p][u] == 0)
				trie[p][u] = ++idx;
			p = trie[p][u];
		}
		cnt[p]++;
	}

	public boolean search(String word)
	{
		int p = 0;
		for (int i = 0; i < word.length(); i++)
		{
			int u = word.charAt(i) - 'a';
			if (trie[p][u] == 0)
				return false;
			p = trie[p][u];
		}
		return cnt[p] != 0;

	}

	public boolean startsWith(String prefix)
	{
		int p = 0;
		for (int i = 0; i < prefix.length(); i++)
		{
			int u = prefix.charAt(i) - 'a';
			if (trie[p][u] == 0)
				return false;
			p = trie[p][u];
		}
		return true;
	}
}
相关推荐
hez201015 小时前
Runtime Async - 步入高性能异步时代
c#·.net·.net core·clr
聚客AI1 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v1 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工1 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
mudtools1 天前
.NET驾驭Word之力:玩转文本与格式
c#·.net
骑自行车的码农1 天前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了1 天前
AcWing学习——双指针算法
c++·算法
moonlifesudo1 天前
322:零钱兑换(三种方法)
算法
唐青枫1 天前
C#.NET 数据库开发提速秘籍:SqlSugar 实战详解
c#·.net
NAGNIP2 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法