力扣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;
	}
}
相关推荐
haogexiaole3 分钟前
Dijkstra 算法
算法
Hello.Reader5 分钟前
从零到一上手 Protocol Buffers用 C# 打造可演进的通讯录
java·linux·c#
浪扼飞舟1 小时前
c#基础(一)
开发语言·c#
papership1 小时前
【入门级-算法-6、排序算法: 插入排序】
数据结构·算法·排序算法
HAH-HAH1 小时前
【蓝桥杯 2024 国 Java A】粉刷匠小蓝
c++·学习·数学·算法·职场和发展·蓝桥杯·组合数学
hweiyu003 小时前
C++设计模式,高级开发,算法原理实战,系统设计与实战(视频教程)
c++·算法·设计模式
一个帅气昵称啊3 小时前
C# .NET EFCore 性能优化
性能优化·c#·.net
大千AI助手3 小时前
粒子群优化(PSO)算法详解:从鸟群行为到强大优化工具
人工智能·算法·优化算法·pso·粒子群优化
ArabySide3 小时前
【C#】理解.NET内存机制:堆、栈与装箱拆箱的底层逻辑及优化技巧
c#
我叫汪枫3 小时前
C语言深度入门系列:第十一篇 - 动态内存管理与数据结构:程序世界的高效算法大师
c语言·数据结构·算法