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