LeetCode 208. Implement Trie (Prefix Tree)

如题所示，要求实现字典树的插入，查找单词，和查找前缀的方法。

Implement a trie with insert, search, and startsWith methods.

Note:
You may assume that all inputs are consist of lowercase letters a-z.

Subscribe to see which companies asked this question.

需要自己设置一个节点Node。节点中应当包含以下两个部分：

1 isEnd：bool类型的变量，主要判断该结点是否为一个单词。

2 child[26]：指向Node结点的指针，下标0~25分别指向26个字母。

在插入的时候，需要在单词的最后一个字母处，将isEnd设置为true。

在查找单词的时候，除了确定字典树中包含了这个单词以外，还要确定最后一个字母的结点中isEnd为true。

查找前缀则只需要是否每一个字母都出现在字典树中即可。

代码实现如下：

class Node{
public:
	bool isEnd = false;
	Node* child[26];
};
class Trie {
private:
	Node* root;
public:
    /** Initialize your data structure here. */
    Trie() {
        root = new Node();
    }

    /** Inserts a word into the trie. */
    void insert(string word) {
    	Node *p=root;
    	for(unsigned int i=0;i<word.length();++i){
    		if(p->child[word[i]-'a'] == NULL){
    			Node* temp = new Node();
    			if(i == word.length()-1)
    				temp->isEnd = true;
    			p->child[word[i]-'a']=temp;
    			p=temp;
    		}else{
    			p = p->child[word[i]-'a'];
    			if(i == word.length()-1)
    				p->isEnd = true;
    		}
    	}
    }

    /** Returns if the word is in the trie. */
    bool search(string word) {
    	Node *p= root;
    	for(unsigned int i=0;i<word.length();++i){
    		if(p->child[word[i]-'a'] == NULL)
    			return false;
    		p=p->child[word[i]-'a'];
    	}
    	if(p->isEnd == true)
    		return true;
    	else
    		return false;
    }

    /** Returns if there is any word in the trie that starts with the given prefix. */
    bool startsWith(string prefix) {
    	Node *p = root;
    	for(unsigned int i=0;i<prefix.length();++i){
    		if(p->child[prefix[i]-'a'] == NULL)
    			return false;
    		p = p->child[prefix[i]-'a'];
    	}
    	return true;
    }
};