LeetCode Implement Trie (Prefix Tree)

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;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值