题解 | #牛群定位系统#
牛群定位系统
https://www.nowcoder.com/practice/d808aaa4c13c4d38bff333aa796e6a1e?tpId=354&tqId=10594633&ru=/exam/oj&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param board char字符型二维数组
* @param words string字符串一维数组
* @return string字符串一维数组
*/
public String[] findWords (char[][] board, String[] words) {
// write code here
ArrayList<String> array = new ArrayList<String>();
for (int x = 0; x < words.length; x++) {
if (findWords(board, words[x], 0, 0, 0)) {
array.add(words[x]);
}
}
int len = array.size();
String[] res = new String[len];
for (int i = 0; i < len; i++) {
res[i] = array.get(i);
}
return res;
}
public boolean findWords(char[][] board, String word, int i, int j, int index) {
if (i < 0 || i >= board.length || j < 0 || j >= board[i].length) {
return index == word.length() ? true : false;
}
if (index == word.length()) {
return true;
}
if (board[i][j] == word.charAt(index)) {
// 做出选择
char c = board[i][j];
board[i][j] = '#';
// 四个方向寻找
boolean up = findWords(board, word, i + 1, j, index + 1);
boolean down = findWords(board, word, i - 1, j, index + 1);
boolean right = findWords(board, word, i, j + 1, index + 1);
boolean left = findWords(board, word, i, j - 1, index + 1);
// 撤销选择
board[i][j] = c;
return up || down || left || right;
} else if (board[i][j] == '#') {
return false;
}
if (index != 0) {
// 说明已经使用过相邻字符了,如果还没有找到,则说明不存在
return false;
}
// 下和右方向寻找
boolean down = findWords(board, word, i + 1, j, index);
boolean right = findWords(board, word, i, j + 1, index);
return down || right;
}
}
判断二维数组中是否存在字符串words[x],如果存在就添加到最后结果中,如果不存在就继续向下找
找四个方向会造成栈溢出,所以从左上角指遍历右边和下边两个方向防止无限递归造成栈溢出。
#java##回溯算法##回溯法##递归##数组移动#
