1 Star 0 Fork 0

xiangxiang/LeetCode-NOTES

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
solution.cpp 1.77 KB
一键复制 编辑 原始数据 按行查看 历史
默然 提交于 7年前 . update all algorithms.
class TrieNode
{
public:
// Initialize your data structure here.
TrieNode()
{
sonNode.resize(26,NULL);
isWordFlag = false;
}
TrieNode* findChar(char c)
{
return sonNode[c-'a'];
}
TrieNode* setChar(char c)
{
sonNode[c-'a'] = new TrieNode();
return sonNode[c-'a'];
}
void setWordFlag()
{
isWordFlag = true;
}
bool isWord()
{
return isWordFlag;
}
private:
bool isWordFlag;
vector<TrieNode*> sonNode;
};
class Trie
{
public:
Trie()
{
root = new TrieNode();
}
// Inserts a word into the trie.
void insert(string s)
{
TrieNode *temp;
TrieNode *node = root;
for(int i=0;i<s.size();i++)
{
temp = node->findChar(s[i]);
if(temp == NULL)
node = node->setChar(s[i]);
else
node = temp;
}
node->setWordFlag();
}
// Returns if the word is in the trie.
bool search(string key)
{
TrieNode *node = root;
for(int i=0;i<key.size();i++)
{
node = node->findChar(key[i]);
if(node == NULL)
return false;
}
return node->isWord();
}
// Returns if there is any word in the trie
// that starts with the given prefix.
bool startsWith(string prefix)
{
TrieNode *node = root;
for(int i=0;i<prefix.size();i++)
{
node = node->findChar(prefix[i]);
if(node == NULL)
return false;
}
return true;
}
private:
TrieNode* root;
};
// Your Trie object will be instantiated and called as such:
// Trie trie;
// trie.insert("somestring");
// trie.search("key");
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/xiangxiang920/LeetCode-NOTES.git
git@gitee.com:xiangxiang920/LeetCode-NOTES.git
xiangxiang920
LeetCode-NOTES
LeetCode-NOTES
master

搜索帮助