其他分享
首页 > 其他分享> > 前缀树应用二

前缀树应用二

作者:互联网

设计一个使用单词列表进行初始化的数据结构,单词列表中的单词 互不相同 。 如果给出一个单词,请判定能否只将这个单词中一个字母换成另一个字母,使得所形成的新单词存在于已构建的神奇字典中。实现 MagicDictionary 类:MagicDictionary() 初始化对象。void buildDict(String[] dictionary) 使用字符串数组 dictionary 设定该数据结构,dictionary 中的字符串互不相同。bool search(String searchWord) 给定一个字符串 searchWord ,判定能否只将字符串中 一个 字母换成另一个字母,使得所形成的新字符串能够与字典中的任一字符串匹配。如果可以,返回 true ;否则,返回 false 。

示例:
输入
inputs = [“MagicDictionary”, “buildDict”, “search”, “search”, “search”, “search”]
inputs = [[], [[“hello”, “leetcode”]], [“hello”], [“hhllo”], [“hell”], [“leetcoded”]]
输出
[null, null, false, true, false, false]
解释
MagicDictionary magicDictionary = new MagicDictionary();
magicDictionary.buildDict([“hello”, “leetcode”]);
magicDictionary.search(“hello”); // 返回 False
magicDictionary.search(“hhllo”); // 将第二个 ‘h’ 替换为 ‘e’ 可以匹配 “hello” ,所以返回 True
magicDictionary.search(“hell”); // 返回 False
magicDictionary.search(“leetcoded”); // 返回 False

建立好前缀树以后,采用深度遍历。遍历条件是:j < 26 && !flag,当flag为true时候退出循环,说明只将字符串中一个字母换成另一个字母,使得所形成的新字符串能够与字典中的任一字符串匹配。

class MagicDictionary {
    class TrieNode{
        TrieNode []children;
        boolean isWord;
        public TrieNode(){
            children = new TrieNode[26];
        }
    }
    TrieNode root;
    /** Initialize your data structure here. */
    public MagicDictionary() {
        root = new TrieNode();
    }
    
    public void buildDict(String[] dictionary) {
        for(String word : dictionary){
            TrieNode node = root;
            for(char ch : word.toCharArray()){
                if(node.children[ch - 'a'] == null){
                    node.children[ch - 'a'] = new TrieNode();
                }
                node = node.children[ch - 'a'];
            }
            node.isWord = true;
        }
    }
    
    public boolean search(String searchWord) {
        TrieNode node = root;
        return dfs(root,searchWord,0,0);
    }
    public boolean dfs(TrieNode root,String searchWord,int i,int edit){
        if(root == null){
            return false;
        }
        if(root.isWord && i == searchWord.length() && edit == 1){
            return true;
        }
        if(i < searchWord.length() && edit <= 1){
            boolean flag = false;
            for(int j = 0; j < 26 && !flag; j++){
                int next = j == searchWord.charAt(i) - 'a' ? edit : edit + 1;
                flag = dfs(root.children[j],searchWord,i+1,next);
            }
            return flag;
        }
        return false;
    }
}

/**
 * Your MagicDictionary object will be instantiated and called as such:
 * MagicDictionary obj = new MagicDictionary();
 * obj.buildDict(dictionary);
 * boolean param_2 = obj.search(searchWord);
 */

标签:search,false,searchWord,TrieNode,MagicDictionary,root,应用,前缀
来源: https://blog.csdn.net/hxxx1314/article/details/122016509