剑指offer | 矩阵中的路径

矩阵中的路径

描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则该路径不能再进入该格子。

代码 (Java)

public class Solution {
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str) {
        if (matrix == null || str == null || rows < 1 || cols < 1)
            return false;

        boolean[] visited = new boolean[matrix.length];
        
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if (hasPathCore(matrix, rows, cols, i, j, str, 0, visited)) {
                    return true;
                }
            }
        }
        
        return false;        
    }
    
    public boolean hasPathCore(char[] matrix, int rows, int cols, int i, int j, 
                               char[] str, int index, boolean[] visited) {
        if (index == str.length)
            return true;
        
        boolean hasPath = false;
        if (i >= 0 && i < rows && j >= 0 && j < cols && 
                !visited[i * cols + j]  && matrix[i * cols + j] == str[index]) {
            visited[i * cols + j] = true;
            hasPath = hasPathCore(matrix, rows, cols, i - 1, j, str, index + 1, visited)
                    || hasPathCore(matrix, rows, cols, i, j - 1, str, index + 1, visited)
                    || hasPathCore(matrix, rows, cols, i + 1, j, str, index + 1, visited)
                    || hasPathCore(matrix, rows, cols, i, j + 1, str, index + 1, visited);
            
            if (!hasPath)
                visited[i * cols + j] = false;
        }
        return hasPath;
    }
}