题目
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board = [ ['A','B','C','E'], ['S','F','C','S'], ['A','D','E','E'] ] 给定 word = "ABCCED", 返回 true. 给定 word = "SEE", 返回 true. 给定 word = "ABCB", 返回 false.
分析
一道回溯算法题,想了想就开始写了,只是写的有毛病。之前自己写回溯算法的话,挺吃力的,不过今天也没有多费劲就写出来了,但是写的不对,稍微改了改,嗯,大部分数据是可以通过的,还有些考虑不周全的地方,改了改代码,也可以过去,不过有些数据量比较大的,(那个测试数据也真是恶心)时间超时了,最近写算法总是时间超时。
看了看解题报告,觉得和自己的大概思路一样呀,后来输出一下发现,自己的回溯真真是把所有情况都列了出来啊,然后看解题报告又改了改。
自己的超时代码find函数的返回是void,即使在找到了正确的单词之后,也只是通过一个全局变量标记一下我找到了,之后还是会将所有的情况全部列出,时间上就很长很长了;网上的代码的find的返回类型是boolean,在找到正确的那个单词之后,立马返回了ture,没有后续情况的列出。
虽然后面的测试数据数组很长,单词字符串很长,但是是最好情况,一路return ture。
代码
class Solution {
public static boolean exist(char[][] board, String word) {
if (word.length() > board.length*board[0].length)return false;
boolean[][] boardFlag = new boolean[board.length][board[0].length];
for (int i = 0; i < board.length; i++)
for (int j = 0; j < board[0].length; j++)
if(find(i,j,board,word,boardFlag))return true;
return false;
}
public static boolean find(int i, int j, char[][] board, String word,boolean[][] boardFlag){
if (board[i][j] == word.charAt(0) && !boardFlag[i][j]){
boardFlag[i][j] = true;
if (word.length()-1 == 0 ) return true;
if (j+1 < board[0].length)
if(find(i,j+1,board,word.substring(1,word.length()),boardFlag)) return true;
if (j-1 >= 0)
if(find(i,j-1,board,word.substring(1,word.length()),boardFlag)) return true;
if (i+1 < board.length)
if(find(i+1,j,board,word.substring(1,word.length()),boardFlag)) return true;
if (i-1 >= 0)
if(find(i-1,j,board,word.substring(1,word.length()),boardFlag)) return true;
boardFlag[i][j] = false;
}
return false;
}
}