212. 单词搜索 II

复制代码
https://leetcode.cn/problems/word-search-ii/description/?envType=study-plan-v2&envId=top-interview-150

思路:

  1. 我们可以先将这个单词网格转化成一颗字典树,然后再拿words进行搜索。

  2. 我们可以先将words装换成字典树,然后在在board中选点看能不能走出一条满足的路径来

对于这两个思路其实差不多,但是1的话如果board太大就很容易超时,而且无用的构建会很多(比如有一个5*5的board,理论一条路径长度应该是25,但是如果words中长的单词都只有5的话我们就根本用不到这么长)。

所以权衡来看第二种思路会更好

思路一(确实会超时):

java 复制代码
class Trie {
        Trie[] children;
        boolean isEnd;
        public Trie() {
            children = new Trie[26];
            isEnd = true; // 每个被构造出的节点都是一个单词的结尾
        }
    }
    public List<String> findWords(char[][] board, String[] words) {
        boolean[][] signed = new boolean[board.length][board[0].length];
        Trie root = new Trie();
        for(int i = 0; i < board.length; i++) {
            for(int j = 0; j < board[0].length; j++) {
                signed[i][j] = true;
                if(root.children[board[i][j] - 'a'] == null) {
                    root.children[board[i][j] - 'a'] = new Trie();
                }
                transfer(board, signed, root.children[board[i][j] - 'a'], i, j);
                signed[i][j] = false;
            }
        }
        List<String> ans = new ArrayList<>();
        for(String word : words) {
            if(search(root, word)) {
                ans.add(word);
            }
        }
        return ans;
    }
    public void transfer(char[][] board, boolean[][] signed, Trie root, int x, int y) {
        for(int i = 0; i < 4; i++) {
            int xx = x + dx[i];
            int yy = y + dy[i];
            if(xx >= 0 && xx < board.length && yy >= 0 && yy < board[0].length && !signed[xx][yy]) {
                signed[xx][yy] = true;
                if(root.children[board[xx][yy] - 'a'] == null) root.children[board[xx][yy] - 'a'] = new Trie();
                transfer(board, signed, root.children[board[xx][yy] - 'a'], xx, yy);
                signed[xx][yy] = false;
            }
        }
    }

思路二:

java 复制代码
    class Trie {
        Trie[] children;
        String word; // 如果当前节点是不是一个单词的结尾,那么word=这个单词,否则word=""
        public Trie() {
            children = new Trie[26];
            word = "";
        }
    }
    public List<String> findWords(char[][] board, String[] words) {
        // 将words构建成字典树
        Trie root = new Trie();
        for(String word : words) {
            add(root, word);
        }
        boolean[][] signed = new boolean[board.length][board[0].length];
        HashSet<String> ans = new HashSet<>();
        // 在board中选取点在Tire中走看是否存在路径
        for(int i = 0; i < board.length; i++) {
            for(int j = 0; j < board[0].length; j++) {
                signed[i][j] = true;
                dfs(root, board, signed, ans, i, j);
                signed[i][j] = false;
            }
        }
        return new ArrayList<>(ans);
    }
    
    public void add(Trie root, String word) {
        for(int i = 0; i < word.length(); i++) {
            if(root.children[word.charAt(i) - 'a'] == null) {
                root.children[word.charAt(i) - 'a'] = new Trie();
            }
            root = root.children[word.charAt(i) - 'a'];
        }
        root.word = word;
    }

    /**
     * 深度优先搜索
     * @param root 当前节点
     * @param board 单词网格
     * @param signed 标记数组
     * @param ans 答案
     * @param x 当前位置x坐标
     * @param y 当前位置y坐标
     */
    public void dfs(Trie root, char[][] board, boolean[][] signed, HashSet<String> ans, int x, int y) {
        if(root == null) return;
        if(!Objects.equals(root.word, "")) {
            ans.add(root.word);
        }
        root = root.children[board[x][y] - 'a'];
        for(int i = 0; i < 4; i++) {
            int xx = x + dx[i];
            int yy = y + dy[i];
            if(xx >= 0 && xx < board.length && yy >= 0 && yy < board[0].length && !signed[xx][yy]) {
                char ch = board[xx][yy];
                signed[xx][yy] = true;
                dfs(root, board, signed, ans, xx, yy);
                signed[xx][yy] = false;
            }
        }
        if (root != null && root.word != "") ans.add(root.word);
    }

    public boolean search(Trie root, String word) {
        for(int i = 0; i < word.length(); i++) {
            if(root.children[word.charAt(i) - 'a'] == null) {
                return false;
            }
            root = root.children[word.charAt(i) - 'a'];
        }
        return "".equals(root.word);
    }
相关推荐
ZBritney几秒前
JAVA中的多线程
java
whn19774 分钟前
达梦数据库的整体负载变化查看
java·开发语言·数据库
小满、4 分钟前
RabbitMQ:Fanout、Direct、Topic 交换机、队列声明与消息转换器
java·分布式·消息队列·rabbitmq·spring amqp
xie_pin_an7 分钟前
深入解析 C 语言排序算法:从快排优化到外排序实现
c语言·算法·排序算法
Hcoco_me11 分钟前
机器学习核心概念与主流算法(通俗详细版)
人工智能·算法·机器学习·数据挖掘·聚类
Hcoco_me12 分钟前
嵌入式场景算法轻量化部署checklist
算法
咸鱼加辣13 分钟前
【python面试】Python 的 lambda
javascript·python·算法
檀越剑指大厂13 分钟前
【Idea系列】换行处理
java·ide·intellij-idea
Jerryhut17 分钟前
sklearn函数总结十二 —— 聚类分析算法K-Means
算法·kmeans·sklearn
wanghowie27 分钟前
01.04 Java基础篇|泛型、注解与反射实战
java·开发语言·windows