其他分享
首页 > 其他分享> > LeetCode 0212 Word Search II

LeetCode 0212 Word Search II

作者:互联网

原题传送门

1. 题目描述

2. Solution

1、思路分析
根据题意,需要逐个遍历二维网格中的每一个单元格,然后搜索从该单元格出发的所有路径,找到其中对应words中的单词的路径。因为这是一个回溯的过程,所以有如下算法:
1》遍历二维网格中的所有单元格。
2》深度优先搜索所有从当前正在遍历的单元格出发,由相邻且不重复的单元格组成的路径。因为题目要求同一个单元格内的字母在一个单词中不能被重复使用;所以在深度优先搜索的过程中,每经过一个单元格,都将该单元格的字母临时修改为特殊字符(如 '#'),以避免再次经过该单元格。
3》如果当前路径是words中的单词,则将其添加到结果集中。如果当前路径是words中任意一个单词的前缀,则继续搜索;反之,如果当前路径不是words中任意一个单词的前缀,则剪枝。可以将words中的所有字符串先添加到前缀树中,而后用O(n)的时间复杂度查询当前路径是否为words中任意一个单词的前缀。

2、代码实现

package Q0299.Q0212WordSearchII;

import java.util.ArrayList;
import java.util.List;

/*
    Backtracking + Trie
 */
public class Solution {
    public List<String> findWords(char[][] board, String[] words) {
        List<String> result = new ArrayList<>();
        TrieNode root = buildTrie(words);
        for (int i = 0; i < board.length; i++) {
            for (int j = 0; j < board[0].length; j++) {
                dfs(board, i, j, root, result);
            }
        }
        return result;
    }

    private void dfs(char[][] board, int i, int j, TrieNode p, List<String> result) {
        char c = board[i][j];
        if (c == '#' || p.children[c - 'a'] == null) return;
        p = p.children[c - 'a'];
        if (p.word != null) {       // found one
            result.add(p.word);
            p.word = null;      //  de-duplicate
        }

        board[i][j] = '#';
        if (i > 0) dfs(board, i - 1, j, p, result);
        if (j > 0) dfs(board, i, j - 1, p, result);
        if (i < board.length - 1) dfs(board, i + 1, j, p, result);
        if (j < board[0].length - 1) dfs(board, i, j + 1, p, result);
        board[i][j] = c;
    }

    private TrieNode buildTrie(String[] words) {
        TrieNode root = new TrieNode();
        for (String w : words) {
            TrieNode p = root;
            for (int i = 0; i < w.length(); i++) {
                char c = w.charAt(i);
                if (p.children[c - 'a'] == null)
                    p.children[c - 'a'] = new TrieNode();
                p = p.children[c - 'a'];
            }
            p.word = w;
        }
        return root;
    }
}

class TrieNode {
    public String word;
    public TrieNode[] children = new TrieNode[26];
}

3、复杂度分析
时间复杂度:

标签:Search,TrieNode,单元格,II,0212,board,words,children,result
来源: https://www.cnblogs.com/junstat/p/16341278.html