leetcode笔记_Trie相关问题

208. 实现 Trie (前缀树)

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

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

说明:

• 你可以假设所有的输入都是由小写字母 a-z 构成的。
• 保证所有输入均为非空字符串。

class Trie {
    
     private class Node{
        public boolean isWord;    //记录该Node是不是一个字符串的结尾
        public TreeMap next;  //该Node的下一个Node的Map

        public Node(boolean isWord){
            this.isWord = isWord;
            this.next = new TreeMap<>();
        }
        public Node(){
            this(false);
        }
    }
    
    private Node root;

    /** Initialize your data structure here. */
    public Trie() {
        root = new Node();
    }
    
    /** Inserts a word into the trie. */
    public void insert(String word) {
        Node cur = root;
        for (int i=0; i

 

211. 添加与搜索单词 - 数据结构设计

设计一个支持以下两种操作的数据结构：

void addWord(word)
bool search(word)

search(word) 可以搜索文字或正则表达式字符串，字符串只包含字母 . 或 a-z 。 . 可以表示任何一个字母。

示例:

addWord("bad")
addWord("dad")
addWord("mad")
search("pad") -> false
search("bad") -> true
search(".ad") -> true
search("b..") -> true

说明:

你可以假设所有单词都是由小写字母 a-z 组成的。

思路：这是一个字符串的模式匹配问题

class WordDictionary {
    private class Node{
        public boolean isWord;
        TreeMap next;
        
        public Node(){
            this.isWord = false;
            this.next = new TreeMap<>();
        }
    }
    
    private Node root;

    /** Initialize your data structure here. */
    public WordDictionary() {
        this.root = new Node();
    }
    
    /** Adds a word into the data structure. */
    public void addWord(String word) {
        Node cur = root;
        for(int i=0; i0){
            return search(root, word, 0);
        }
        return false;
    }
    private boolean search(Node node, String word, int index){
        //递归退出条件
        if(index == word.length()){
            return node.isWord;
        }
        
        
        char c = word.charAt(index);
        if(c!='.'){
            if(node.next.get(c)!=null){
                return search(node.next.get(c), word, index+1);  //需要return???
            }
            return false;
        }else{
            for(char key: node.next.keySet()){
                
               //只判断可不可以匹配,不需要计算有多少个相匹配的，所以只要找到一个匹配的就可以返回true
               if(search(node.next.get(key), word, index+1)){   
                   return true;
               }
            }
             return false;          
        }
    }
}

/**
 * Your WordDictionary object will be instantiated and called as such:
 * WordDictionary obj = new WordDictionary();
 * obj.addWord(word);
 * boolean param_2 = obj.search(word);
 */

677. 键值映射

实现一个 MapSum 类里的两个方法，insert 和 sum。

对于方法 insert，你将得到一对（字符串，整数）的键值对。字符串表示键，整数表示值。如果键已经存在，那么原来的键值对将被替代成新的键值对。

对于方法 sum，你将得到一个表示前缀的字符串，你需要返回所有以该前缀开头的键的值的总和。

示例 1:

输入: insert("apple", 3), 输出: Null
输入: sum("ap"), 输出: 3
输入: insert("app", 2), 输出: Null
输入: sum("ap"), 输出: 5

思路：这是一个字符串映射问题，可以通过维护Trie节点内的值来完成 

class MapSum {
    private class Node{
        public int val;
        public TreeMap next;
        
        public Node(){
            this.val = 0;
            next = new TreeMap<>();
        }
    }
    
    private Node root;

    /** Initialize your data structure here. */
    public MapSum() {
        root = new Node();
    }
    
    public void insert(String key, int val) {
        Node cur = root;
        for(int i=0; i