LeetCode 单词搜索

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

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

  • 示例:

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

给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.

算法如下:

// 回溯法
// 时间复杂度: O(m*n*m*n)
// 空间复杂度: O(m*n)
class Solution {

    private int m; 
    private int n;
    private int[][] d = {{-1,0}, {0,1}, {1,0}, {0,-1}};
    private boolean[][] visited;

    public boolean exist(char[][] board, String word) {

        if(board == null || word == null)
           throw new IllegalArgumentException("board or word can not be null!");

        m = board.length;
        if(m == 0)
            throw new IllegalArgumentException("board can not be empty.");
        n = board[0].length;
        if(n == 0)
            throw new IllegalArgumentException("board can not be empty.");

        visited = new boolean[m][n];

        for(int i=0; i<m; i++)
            for(int j=0; j<n; j++)
                if(searchWord(board, word, 0, i, j))
                    return true;

        return false;
    }

    // 从 boart[startx][starty] 处开始找 word[index...word.length()) 
    private boolean searchWord(char[][] board, String word, int index, 
                               int startx, int starty){

        // 终止条件
        if(index == word.length() - 1)
            return board[startx][starty] == word.charAt(index);

        if(board[startx][starty] == word.charAt(index)){
            visited[startx][starty] = true;
            // 从 startx,starty出发, 向四个方向寻找
            for(int i=0; i<4; i++){
                int newx = startx + d[i][0];
                int newy = starty + d[i][1];
                if( inArea(newx, newy) && !visited[newx][newy] &&
                   searchWord(board , word, index+1, newx, newy) )
                        return true;
            }
            visited[startx][starty] = false; // 回溯
        }
        return false;
    }

    // 判断是否越界
    private boolean inArea(int x, int y){
        return x >= 0 && x < m && y >= 0 && y < n; 
    }
} 

猜你喜欢

转载自blog.csdn.net/annotation_yang/article/details/81436510
今日推荐