208. 实现 Trie (前缀树)

Trie(发音类似 "try")或者说 前缀树 是一种树形数据结构,用于高效地存储和检索字符串数据集中的键。这一数据结构有相当多的应用情景,例如自动补完和拼写检查。

请你实现 Trie 类:

  • Trie() 初始化前缀树对象。
  • void insert(String word) 向前缀树中插入字符串 word 。
  • boolean search(String word) 如果字符串 word 在前缀树中,返回 true(即,在检索之前已经插入);否则,返回 false 。
  • boolean startsWith(String prefix) 如果之前已经插入的字符串 word 的前缀之一为 prefix ,返回 true ;否则,返回 false 。

示例:

输入
["Trie", "insert", "search", "search", "startsWith", "insert", "search"]
[[], ["apple"], ["apple"], ["app"], ["app"], ["app"], ["app"]]
输出
[null, null, true, false, true, null, true]

解释
Trie trie = new Trie();
trie.insert("apple");
trie.search("apple");   // 返回 True
trie.search("app");     // 返回 False
trie.startsWith("app"); // 返回 True
trie.insert("app");
trie.search("app");     // 返回 True

提示:

  • 1 <= word.length, prefix.length <= 2000
  • word 和 prefix 仅由小写英文字母组成
  • insertsearch 和 startsWith 调用次数 总计 不超过 3 * 104 次

解题思路

  1. 用到字典树和前缀树的思想
  2. 字典树指的是结点包含一个子结点数组,用一个标记判断是否为一个单词,
    1. 如果标记为false说明从根节点到当前结点只是一个完整单词的前缀的一部分
    2. 如果标记为true说明从根节点到当前结点存在一个完整的单词
  3. 用到字符减去字符'a'得到该字符的相对索引值再为该索引下的数组元素new 一个 Trie对象用于后面判断该字符是否存在
    1. 若该字符转化为索引对应的数组元素值不为空说明该字符存在
    2. 否则说明该字符不存在
class Trie {
   private Trie[] children;
   private boolean isEnd;
    public Trie(){
        this.children=new Trie[26];
    }
    public void insert(String word) {
        Trie head=this;
        int index=-1;
        for(int i=0;i<word.length();i++){
            index=word.charAt(i)-'a';
            if(head.children[index]==null){
                head.children[index]=new Trie();
            }
            head=head.children[index];
        }
          head.isEnd=true;
    }
    public boolean search(String word) {
        Trie head=this;
         int index=-1;
        for(int i=0;i<word.length();i++){
             index=word.charAt(i)-'a';
            if(head.children[index]==null)
                return false;
            head=head.children[index];
        }
        if(head.isEnd)
            return true;
        return false;
        
    }
    public boolean startsWith(String prefix) {
            Trie head=this;     
            int index=-1;
          for(int i=0;i<prefix.length();i++){
               index=prefix.charAt(i)-'a';
            if(head.children[index]==null)
                return false;
            head=head.children[index];
          }
          return true;
    }
}

/**
 * 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);
 */

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,以下是使用 Kotlin 实现 Trie 的代码: ```kotlin class Trie() { private val root = TrieNode() fun insert(word: String) { var cur = root for (c in word) { if (!cur.children.containsKey(c)) { cur.children[c] = TrieNode() } cur = cur.children[c]!! } cur.isEnd = true } fun search(word: String): Boolean { var cur = root for (c in word) { if (!cur.children.containsKey(c)) { return false } cur = cur.children[c]!! } return cur.isEnd } fun startsWith(prefix: String): Boolean { var cur = root for (c in prefix) { if (!cur.children.containsKey(c)) { return false } cur = cur.children[c]!! } return true } } class TrieNode { val children = mutableMapOf<Char, TrieNode>() var isEnd = false } ``` 在上面的代码中,我们定义了 `Trie` 类和 `TrieNode` 类,其中 `TrieNode` 表示前缀树的节点,包含一个表示是否为单词结尾的布尔值和一个 `children` 字典,用于存储子节点。`Trie` 类中包含 `insert`、`search` 和 `startsWith` 三个方法,分别用于插入、查找和判断是否存在前缀。在 `insert` 方法中,我们遍历字符串的每个字符,如果当前节点的子节点中没有该字符,则新建一个 `TrieNode` 并加入子节点中。在遍历结束后,将最后一个节点标记为单词结尾。在 `search` 和 `startsWith` 方法中,我们同样遍历字符串的每个字符,如果当前节点的子节点中没有该字符,则直接返回 false。如果遍历结束后,当前节点是单词结尾,则说明该单词存在,返回 true;否则说明该单词不存在,返回 false。在 `startsWith` 方法中,我们不需要判断当前节点是否为单词结尾,只需判断该前缀是否存在即可。 希望能对你有所帮助!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值