LeetCode Implement Trie (Prefix Tree)

最新推荐文章于 2024-03-22 00:51:13 发布

原创最新推荐文章于 2024-03-22 00:51:13 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #字典树 #前缀树

LeetCode 专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了一种高效的数据结构——Trie树的实现方法。Trie树，又称前缀树，是一种用于存储字符串集合的树形数据结构，能够快速进行字符串的插入、查找和前缀匹配操作。文章通过具体的C++代码示例，展示了如何创建Trie树节点，以及如何通过插入、搜索和检查前缀功能来操作Trie树。

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

class Trie {
	class Node{
		Node() :next(26, nullptr), m_IsEnd(false){
		}

		bool  m_IsEnd;
		vector<Node*> next;
	};
public:
	/** Initialize your data structure here. */
	Trie() {

	}

	/** Inserts a word into the trie. */
	void insert(string word) {
		Node* tn = &m_Head;
		int c;
		for (auto i : word) {
			//c = word2Int(i);
			c = i - 'a';
			if (tn->next[c] == nullptr) {
				tn->next[c] = new Node();
			}
			tn = (tn->next[c]);
		}
		tn->m_IsEnd = true;
	}

	/** Returns if the word is in the trie. */
	bool search(string word) {
		Node *tn = &m_Head;
		int c;
		for (auto i : word) {
			//c = word2Int(i);
			c = i - 'a';
			if (tn->next[c] == nullptr) {
				return false;
			}
			tn = tn->next[c];
		}
		return tn->m_IsEnd;
	}

	/** Returns if there is any word in the trie that starts with the given prefix. */
	bool startsWith(string prefix) {
		Node *tn = &m_Head;
		int c;
		for (auto i : prefix) {
			//c = word2Int(i);
			c = i - 'a';
			if (tn->next[c] == nullptr) {
				return false;
			}
			tn = tn->next[c];
		}
		return true;
	}

private:
	inline int word2Int(char c) {
		return c - 'a';
	}
private:
	Node m_Head;
};