力扣刷题记录 -- JAVA--53---208. 实现 Trie (前缀树)

最新推荐文章于 2025-01-20 11:37:07 发布

@白圭

最新推荐文章于 2025-01-20 11:37:07 发布

阅读量54

点赞数 1

分类专栏： # 力扣习题记录册__JAVA__热题100 文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/taiyuezyh/article/details/133902618

版权

力扣习题记录册__JAVA__热题100 专栏收录该内容

63 篇文章 0 订阅

订阅专栏

一、题目

在这里插入图片描述

二、代码

class Trie 
{
    //有前缀和有起始
    //前缀树就是26叉树   同时确认自身是否可以终止
    //节点类有两个属性  next 和 isend

    class Node
    {
       private boolean isEnd ;
       private Node[] next;
       public Node()
       {
            isEnd = false;
            next = new Node[26];
       }
    }

    private Node root;
    public Trie()
    {
        root = new Node();
    }
    
    public void insert(String word) 
    {
          Node node = root;
          for(char i : word.toCharArray())
          {
              if(node.next[i-'a'] == null)
              {
                  node.next[i-'a'] = new Node();
              }
              node = node.next[i-'a'];
          }
          node.isEnd = true;
    }
    
    public boolean search(String word) 
    {
         Node node = mySearch(word);
         return node!=null && node.isEnd == true;
    }
    
    //简化版search
    public boolean startsWith(String prefix) 
    {
        Node node = mySearch(prefix);
        return node!=null ;
    }

    public Node mySearch(String word)
    {
        Node node = root;
        for(char i: word.toCharArray())
        {
            if(node.next[i-'a'] == null)
            {
                return null;
            }
            node = node.next[i-'a'];
        }
        return node;
    }
}

/**
 * Your Trie object will be instantiated and called as such:
 * Trie obj = new Trie();
 * obj.insert(word);
 * boolean param_2 = obj.search(word);
 * boolean param_3 = obj.startsWith(prefix);
 */