leetcode Implement Trie (Prefix Tree)

最新推荐文章于 2024-03-22 00:51:13 发布

原创最新推荐文章于 2024-03-22 00:51:13 发布 · 312 阅读

0 ·

CC 4.0 BY-SA版权

leetCode 专栏收录该内容

199 篇文章

订阅专栏

本文详细介绍了Trie树（字典树）的数据结构实现及核心方法：插入(insert)、查找(search)与前缀匹配(startsWith)。通过具体的代码示例展示了如何使用Trie树高效地存储和检索字符串。

题目链接

思路：
无

class TrieNode {
    // Initialize your data structure here.
    TrieNode charecters[];
    boolean end;
    public TrieNode() {
        charecters=new TrieNode[26];
        end=false;
    }
}

public class Trie {
    private TrieNode root;

    public Trie() {
        root = new TrieNode();
        root.charecters=new TrieNode[26]; 
    }

    // Inserts a word into the trie.
    public void insert(String word) {
        int n=word.length();
        TrieNode temp=root;
        for(int i=0;i<n;i++)
        {

            if(temp.charecters[word.charAt(i)-'a']==null)
            {
                temp.charecters[word.charAt(i)-'a']=new TrieNode();
            }
            temp=temp.charecters[word.charAt(i)-'a'];

        }
        temp.end=true;
    }

    // Returns if the word is in the trie.
    public boolean search(String word) {
         int n=word.length();
         TrieNode temp=root;
         for(int i=0;i<n;i++)
         {

            if(temp.charecters[word.charAt(i)-'a']==null)
            {
                return false;
            }
            temp=temp.charecters[word.charAt(i)-'a'];

         }
         return temp.end;
    }

    // Returns if there is any word in the trie
    // that starts with the given prefix.
    public boolean startsWith(String prefix) {
        int n=prefix.length();
        TrieNode temp=root;
        for(int i=0;i<n;i++)
        {

            if(temp.charecters[prefix.charAt(i)-'a']==null)
            {
                return false;
            }
            temp=temp.charecters[prefix.charAt(i)-'a'];

        }
        if(temp.end)
        {
            return true;
        }
        for(int i=0;i<26;i++)
        {
            if(temp.charecters[i]!=null)
            {
                return true;
            }
        }

        return false;
    }
}

// Your Trie object will be instantiated and called as such:
// Trie trie = new Trie();
// trie.insert("somestring");
// trie.search("key");