爱程序网

LeetCode - Word Search

来源: 阅读:

题目:

Given a 2D board and a word, find if the word exists in the grid.

The word can be constructed from letters of sequentially adjacent cell, where "adjacent" cells are those horizontally or vertically neighboring. The same letter cell may not be used more than once.

For example,
Given board =

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

word = "ABCCED", -> returns true,
word = "SEE", -> returns true,
word = "ABCB", -> returns false.

思路:

递归;及时return true,减少计算量

package recursion;

public class WordSearch {

    public boolean exist(char[][] board, String word) {
        int m;
        int n;
        int wordLen;
        if (board == null || word == null || 
                (m = board.length) == 0 || (n = board[0].length) == 0 || 
                (wordLen = word.length()) == 0 || wordLen > m * n) 
            return false;
        boolean[][] visited = new boolean[m][n];

        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                if (dfs(board, i, j, visited, word, 0, m, n, wordLen))
                    return true;
            }
        }
        return false;
    }
 
    private boolean dfs(char[][] board, int x, int y, boolean[][] visited, String word, int pos, int m, int n, int wordLen) {
        if (visited[x][y] || board[x][y] != word.charAt(pos)) return false;
        if (pos == wordLen - 1) return true;
        visited[x][y] = true;

        if (x - 1 >= 0 && dfs(board, x - 1, y, visited, word, pos + 1, m, n, wordLen)) return true;
        
        if (x + 1 < m && dfs(board, x + 1, y, visited, word, pos + 1, m, n, wordLen)) return true;
        
        if (y - 1 >= 0 && dfs(board, x, y - 1, visited, word, pos + 1, m, n, wordLen)) return true;
        
        if (y + 1 < n && dfs(board, x, y + 1, visited, word, pos + 1, m, n, wordLen)) return true;
        
        visited[x][y] = false;
        
        return false;
    }
    
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        char[][] board = 
                        {
                            {'A','B','C','E'},
                            {'S','F','C','S'},
                            {'A','D','E','E'}
                        };
        WordSearch w = new WordSearch();
        System.out.println(w.exist(board, "ABCCED"));
        System.out.println(w.exist(board, "SEE"));
        System.out.println(w.exist(board, "ABCB"));
    }

}

 

关于爱程序网 - 联系我们 - 广告服务 - 友情链接 - 网站地图 - 版权声明 - 人才招聘 - 帮助