Bootstrap

leetcode79. 单词搜索(回溯算法)

给定一个二维网格和一个单词,找出该单词是否存在于网格中。

单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

示例:

board =
[
[‘A’,‘B’,‘C’,‘E’],
[‘S’,‘F’,‘C’,‘S’],
[‘A’,‘D’,‘E’,‘E’]
]

给定 word = “ABCCED”, 返回 true
给定 word = “SEE”, 返回 true
给定 word = “ABCB”, 返回 false

代码

class Solution {
    boolean[][] vis;
    public boolean exist(char[][] board, String word) {
        int[][] dir = new int[][]{{-1, 0}, {1, 0}, {0, 1}, {0, -1}};
        int n=board.length,m=board[0].length;

        vis=new boolean[n][m];
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
            {
                if(checkExist(board,word,dir,0,i,j))
                   return true;
            }
        return false;
    }
    public boolean checkExist(char[][] board, String word, int[][] dir,int loc,int x,int y) {

        if(board[x][y]!=word.charAt(loc))
            return false;
        if(loc==word.length()-1) return true;
        vis[x][y]=true;//标记

        for(int[] d:dir)
        {
            int nextX=x+d[0],nextY=y+d[1];
            if (nextX < 0 || nextX >= board.length || nextY < 0 || nextY >= board[0].length||vis[nextX][nextY])
                continue;
            if(checkExist(board, word, dir, loc+1, nextX, nextY)) return true;
        }
        vis[x][y]=false;//回溯
        return false;
    }
}
;