【LeetCode】208.实现Trie(前缀树)

最新推荐文章于 2024-03-24 13:00:11 发布

原创最新推荐文章于 2024-03-24 13:00:11 发布 · 260 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#字典树

LeetCode 专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一个Trie(前缀树)的数据结构实现，包括insert、search和startsWith三个核心操作，通过示例展示了如何使用这个数据结构进行单词插入、搜索和前缀查找。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目难度★★★

题目描述

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert("apple");
trie.search("apple");   // 返回 true
trie.search("app");     // 返回 false
trie.startsWith("app"); // 返回 true
trie.insert("app");   
trie.search("app");     // 返回 true

题目地址 https://leetcode-cn.com/problems/implement-trie-prefix-tree

实现代码

class Trie {
public:
	/** Initialize your data structure here. */
	Trie(){
		root = new TrieNode();
	}

	/** Inserts a word into the trie. */
	void insert(string word) {
		TrieNode* p = root;
		for (auto& a : word) {
			int i = a - 'a';
			if (!p->child[i])
				p->child[i] = new TrieNode();
			p = p->child[i];
		}
		p->isWord = true;
	}

	/** Returns if the word is in the trie. */
	bool search(string word) {
		TrieNode* p = root;
		for (auto& a : word) {
			int i = a - 'a';
			if (!p->child[i])
				return false;
            p = p->child[i];
		}
		return p->isWord;
	}

	/** Returns if there is any word in the trie that starts with the given prefix. */
	bool startsWith(string prefix) {
		TrieNode* p = root;
		for (auto& a : prefix) {
			int i = a - 'a';
			if (!p->child[i])
				return false;
			p = p->child[i];
		}
		return true;
	}
private:
	struct TrieNode
	{
		TrieNode():isWord(false){
			for (auto &a:child)
			{
				a = NULL;
			}
		}

		TrieNode* child[26];
		bool isWord;
	};

	TrieNode* root;
};

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