【LeetCode/LintCode】題解丨 Google面試題：單詞搜尋 II
阿新 • • 發佈：2020-09-23
給出一個由小寫字母組成的矩陣和一個字典。找出所有同時在字典和矩陣中出現的單詞。一個單詞可以從矩陣中的任意位置開始，可以向左/右/上/下四個相鄰方向移動。一個字母在一個單詞中只能被使用一次。且字典中不存在重複單詞。
線上評測地址：領釦刷題地址
樣例 1:
樣例 1:
輸入：["doaf","agai","dcan"]，["dog","dad","dgdg","can","again"]
輸出：["again","can","dad","dog"]
解釋：
d o a f
a g a i
d c a n
矩陣中查詢，返回 ["again","can","dad","dog"]。
樣例 2:
輸入：["a"]，["b"]
輸出：[]
解釋：
a
矩陣中查詢，返回 []。
題解
使用 HashMap 的版本。 將所有的單詞的 Prefix 都加入 HashMap 中。HashMap 的 value 使用者判斷這是一個 prefix 還是一個 word。 如果是 prefix 就是 false，如果是 word 就是 true.
public class Solution {
public static int[] dx = {0, 1, -1, 0};
public static int[] dy = {1, 0, 0, -1};
/** * @param board: A list of lists of character * @param words: A list of string * @return: A list of string */
public List<String> wordSearchII(char[][] board, List<String> words) {
if (board == null || board.length == 0) {
return new ArrayList<>();
}
if (board[0] == null || board[0].length == 0) {
return new ArrayList<>();
}
boolean[][] visited = new
Map<String, Boolean> prefixIsWord = getPrefixSet(words);
Set<String> wordSet = new HashSet<>();
for (int i = 0; i < board.length; i++) {
for (int j = 0; j < board[i].length; j++) {
visited[i][j] = true;
dfs(board, visited, i, j, String.valueOf(board[i][j]), prefixIsWord, wordSet);
visited[i][j] = false;
}
}
return new ArrayList<String>(wordSet);
}
private Map<String, Boolean> getPrefixSet(List<String> words) {
Map<String, Boolean> prefixIsWord = new HashMap<>();
for (String word : words) {
for (int i = 0; i < word.length() - 1; i++) {
String prefix = word.substring(0, i + 1);
if (!prefixIsWord.containsKey(prefix)) {
prefixIsWord.put(prefix, false);
}
}
prefixIsWord.put(word, true);
}
return prefixIsWord;
}
private void dfs(char[][] board,
boolean[][] visited,
int x,
int y,
String word,
Map<String, Boolean> prefixIsWord,
Set<String> wordSet) {
if (!prefixIsWord.containsKey(word)) {
return;
}
if (prefixIsWord.get(word)) {
wordSet.add(word);
}
for (int i = 0; i < 4; i++) {
int adjX = x + dx[i];
int adjY = y + dy[i];
if (!inside(board, adjX, adjY) || visited[adjX][adjY]) {
continue;
}
visited[adjX][adjY] = true;
dfs(board, visited, adjX, adjY, word + board[adjX][adjY], prefixIsWord, wordSet);
visited[adjX][adjY] = false;
}
}
private boolean inside(char[][] board, int x, int y) {
return x >= 0 && x < board.length && y >= 0 && y < board[0].length;
}
}
更多題解參考：九章官網Solution