leetcode 208. 实现 Trie (前缀树) Trie

本文介绍了一种Trie(前缀树)的数据结构实现,包括insert(插入)、search(搜索)和startsWith(前缀查找)三个核心方法。通过递归地构建节点并使用标志位来标记完整字符串的位置。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

实现一个 Trie (前缀树),包含 insert, search, 和 startsWith 这三个操作。
emmmm….基本操作,用一个flag记录当前位置是否是一个完整串或者直接判断String是否为null.

class Trie {
        Trie[] next;
        String word=null; //判断是否是一个完整的串
        /** Initialize your data structure here. */
        public Trie() {
            next=new Trie[26];
        }

        /** Inserts a word into the trie. */
        public void insert(String word) {
            if(word==null||word.equals("")){
                return;
            }
            Trie cur=this;
            for(char ch:word.toCharArray()){  //遍历
                if(cur.next[ch-'a']==null){
                    cur.next[ch-'a']=new Trie();
                }
                cur=cur.next[ch-'a'];
            }
            cur.word=word; //设置串的识别标志
        }

        /** Returns if the word is in the trie. */
        public boolean search(String word) {
            Trie cur=this;
            for(char ch:word.toCharArray()){
                if(cur.next[ch-'a']==null){  //说明没有这个串 直接放回
                    return false;
                }
                cur=cur.next[ch-'a'];
            }
            if(cur.word==null){  //只是前缀 不是一个完整串
                return false;
            }
            return true;
        }

        /** Returns if there is any word in the trie that starts with the given prefix. */
        public boolean startsWith(String prefix) {
            Trie cur=this;
            for(char ch:prefix.toCharArray()){
                if(cur.next[ch-'a']==null){
                    return false;
                }
                cur=cur.next[ch-'a'];
            }
            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);
 */
### 实现 Trie前缀) 为了实现 Trie 数据结构,在 Java 中可以定义一个 `TrieNode` 类来表示节点,并创建相应的操作方法。下面是一个完整的解决方案: #### 定义 Trie 节点 每个节点包含指向其子节点的链接列表以及是否为单词结束标志。 ```java class TrieNode { private final int R = 26; private TrieNode[] links; private boolean isEnd; public TrieNode() { links = new TrieNode[R]; } // 判断是否存在字符对应的子节点 public boolean containsKey(char ch) { return links[ch - 'a'] != null; } // 获取指定字符对应的孩子结点 public TrieNode get(char ch) { return links[ch - 'a']; } // 设置当前结点孩子结点 public void put(char ch, TrieNode node) { links[ch - 'a'] = node; } // 标记该位置是否有字符串终止于此处 public void setEnd() { isEnd = true; } // 是否有字符串在此终结 public boolean isEnd() { return isEnd; } } ``` #### 创建 Trie 类并提供接口函数 通过封装上述基本功能,可构建更高级别的 API 来支持插入、搜索查询以特定前缀开头的所有键等功能。 ```java class Trie { private final TrieNode root; /** Initialize your data structure here. */ public Trie() { root = new TrieNode(); } /** Inserts a word into the trie. */ public void insert(String word) { TrieNode node = root; for (int i = 0; i < word.length(); i++) { char currentChar = word.charAt(i); if (!node.containsKey(currentChar)) { node.put(currentChar, new TrieNode()); } node = node.get(currentChar); } node.setEnd(); } /** Returns if the word is in the trie. */ public boolean search(String word) { TrieNode node = searchPrefix(word); return node != null && node.isEnd(); } /** Returns if there is any word in the trie that starts with the given prefix. */ public boolean startsWith(String prefix) { TrieNode node = searchPrefix(prefix); return node != null; } private TrieNode searchPrefix(String word) { TrieNode node = root; for (int i = 0; i < word.length(); i++) { char curLetter = word.charAt(i); if (node.containsKey(curLetter)) { node = node.get(curLetter); } else { return null; } } 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); */ ``` 此代码实现Trie 的核心逻辑,包括初始化、插入新词、查找完整匹配项以及检查给定前缀的存在性[^1]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值