力扣208.实现前缀树

阳光男孩01

于 2024-07-29 10:26:46 发布

阅读量254

点赞数 3

文章标签： leetcode c# 算法

本文链接：https://blog.csdn.net/Pisasama/article/details/140764735

版权

力扣208.实现前缀树

前缀树是一个每个节点存一个字母的树
- 每个节点最多有26棵子树(26个字母)

  class Trie {
      bool isend;
      Trie* next[26];
  public:
      Trie() {
          isend = false;
          memset(next,0,sizeof(next));
      }
      
      void insert(string word) {
          Trie* node = this;
          for(char c:word)
          {
              if(node->next[c-'a'] == NULL)
                  node->next[c-'a'] = new Trie();
              node = node->next[c-'a'];
          }
          node->isend = true;
      }
      
      bool search(string word) {
          Trie* node = this;
          for(char c:word)
          {
              node = node->next[c-'a'];
              if(node == NULL)
                  return false;
          }
          return node->isend;
      }
      
      bool startsWith(string prefix) {
          Trie* node = this;
          for(char c:prefix)
          {
              node = node->next[c-'a'];
              if(node == NULL)
                  return false;
          }
          return true;
      }
  };