剑指offer第六十五题矩阵中的路径

题目描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则该路径不能再进入该格子。 例如 矩阵中包含一条字符串"bcced"的路径，但是矩阵中不包含"abcb"和"absf"路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

方法一：回溯法

回溯法基本思想：

0.根据给定数组，初始化一个标志位数组，初始化为false，表示未走过，true表示已经走过，不能走第二次

1.根据行数和列数，遍历数组，先找到一个与str字符串的第一个元素相匹配的矩阵元素，进入judge

2.根据i和j先确定一维数组的位置，因为给定的matrix是一个一维数组

3.确定递归终止条件：越界，当前找到的矩阵值不等于数组对应位置的值，已经走过的，这三类情况，都直接false，说明这条路不通

4.若k，就是待判定的字符串str的索引已经判断到了最后一位，此时说明是匹配成功的

5.下面就是本题的精髓，递归不断地寻找周围四个格子是否符合条件，只要有一个格子符合条件，就继续再找这个符合条件的格子的四周是否存在符合条件的格子，直到k到达末尾或者不满足递归条件就停止。

6.走到这一步，说明本次是不成功的，我们要还原一下标志位数组index处的标志位，进入下一轮的判断。

dfs的模板（这类题不用判断边界条件了）：

boolean dfs(){
    // 第1步：检查是否被访问过，或者是否满足当前限制条件，递归终止条件（边界条件+限制条件）
       return false;

    // 第2步：检查是否满足返回结果条件（终止满足条件）
	   return true;

    // 第3步：标记走过的路程
	   flag[index] = true;

    // 第4步：都没有返回，说明应该进行下一步递归
       if(dfs(下一次))
       return true;

    // 第5步：此路不通，还原，返回false
       flag[index] = false;

    // 第6步：此路不通，还原，返回false
       return false;
}  
main() {
    flag = false;
    for (对所有可能情况) {
        return dfs();
    }
    return false;
}

class Solution {
    public boolean hasPath(char[] matrix, int rows, int cols, char[] str)
    {
        //标志位，初始化为false
        boolean[] flag = new boolean[matrix.length];
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                //循环遍历二维数组，找到起点等于str第一个元素的值，再递归判断四周是否有符合条件的----回溯法
                if(judge(matrix,i,j,rows,cols,flag,str,0)){
                    return true;
                }
            }
        }
        return false;
    }

    //judge(初始矩阵，索引行坐标i，索引纵坐标j，矩阵行数，矩阵列数，待判断的字符串，字符串索引初始为0即先判断字符串的第一位)
    private boolean judge(char[] matrix,int i,int j,int rows,int cols,boolean[] flag,char[] str,int k){
        //先根据i和j计算匹配的第一个元素转为一维数组的位置，cols是有几列，也就是一行多少个的意思。
        int index = i*cols+j;
        //递归终止条件
        if(i<0 || j<0 || i>=rows || j>=cols || matrix[index] != str[k] || flag[index] == true)
            return false;
        //若k已经到达str末尾了，说明之前的都已经匹配成功了，直接返回true即可
        if(k == str.length-1)//果然标黄是有原因的,这里写成k == str.length就标黄了
            return true;
        //要走的第一个位置置为true，表示已经走过了
        flag[index] = true;

        //回溯，递归寻找，每次找到了就给k加一，找不到，还原
        if(judge(matrix,i-1,j,rows,cols,flag,str,k+1) ||
                judge(matrix,i+1,j,rows,cols,flag,str,k+1) ||
                judge(matrix,i,j-1,rows,cols,flag,str,k+1) ||
                judge(matrix,i,j+1,rows,cols,flag,str,k+1)  )
        {
            return true;
        }
        //走到这，说明这一条路不通，还原，再试其他的路径
        flag[index] = false;
        return false;
    }
}

public class Main {
    public static void main(String[] args) {
        char[] temp = {'a','b','c','e','s','f','c','s','a','d','e','e'};
        char[] str = {'a','b','f','s'};
        Solution p = new Solution();
        boolean b = p.hasPath(temp,3,4,str);
        System.out.println("输出结果"+ b);
    }
}

或者：

import java.util.*;

class Solution {
    public boolean exist(char[][] board, String word) {
        char[] words = word.toCharArray();
        for(int i = 0; i < board.length; i++) {
            for(int j = 0; j < board[0].length; j++) {
                if(dfs(board, words, i, j, 0)) {
                    return true;
                }
            }
        }
        return false;
    }
    boolean dfs(char[][] board, char[] word, int i, int j, int k) {
        if(i >= board.length || i < 0 || j >= board[0].length || j < 0 || board[i][j] != word[k]) {
            return false;
        }
        if(k == word.length - 1) {
            return true;
        }
        //标记走过
        board[i][j] = '\0';
        boolean res = dfs(board, word, i + 1, j, k + 1) || dfs(board, word, i - 1, j, k + 1) ||
                dfs(board, word, i, j + 1, k + 1) || dfs(board, word, i , j - 1, k + 1);
        //标记未走过
        board[i][j] = word[k];
        return res;
    }
}
public class Main {
    public static void main(String[] args) {
        char[][] board = {{'A','B','C','E'},{'S','F','C','S'},{'A','D','E','E'}};
        String word = "ABCCED";
        Solution s = new Solution();
        boolean res = s.exist(board, word);
        System.out.println("结果：" + res);
    }
}

方法二：使用栈（暂时搁置）

package test;
import java.util.Stack;
class Solution {
    private static int col;
    public static boolean hasPath(char[] matrix, int rows, int cols, char[] str){
        col = cols;
        if(matrix.length==0||str.length==0||str.length>matrix.length)
            return false;
        for (int total = 0; total < matrix.length; total++) {
            if (matrix[total] == str[0]) {   //只有第一个元素相同时，才调用方法压栈
                if (checkpath(matrix, total, str)) return true;
            }
        }
        return false;
    }
    private static boolean checkpath(char[] matrix, int in, char[] str) {
        if(str.length==1) return matrix[in]==str[0];
        Stack<Integer> sta = new Stack<Integer>();
        sta.push(in);
        int k = 1, i, j, index;
        int back[] = new int[matrix.length];//使用一个辅助数组
        back[in] = 1; //in是传入的位置，将首位置标记置1（back是int数组）
        boolean findflag;
        while (!sta.empty()) { //栈为空时，一次搜寻结束，只有首元素相同时，才入栈
//这是通过两层for循环控制的
            index = sta.peek();
            findflag = true;//第一个元素找到了，所以初始的标志true
            j=index%col;   //j是第几列
            i=(index-j)/col; //i是第几行
            if ((index - col >= 0) && //上
                    back[index - col] == 0 &&
                    matrix[index - col] == str[k]) {
                index -= col;
            } else if ((index + 1 < (i + 1) * col) &&//右
                    (back[index + 1]) == 0 &&
                    matrix[index + 1] == str[k]) {
                index += 1;
            } else if ((index + col < matrix.length)//下
                    && (back[index + col] == 0)
                    && matrix[index + col] == str[k]) {
                index += col;
            } else if ((index - 1 >= i * col) &&//左
                    (back[index - 1] == 0) &&
                    matrix[index - 1] == str[k]) {
                index -= 1;
            } else {
                findflag = false;//没有找到
            }
            if (findflag && back[index] == 0) {
//一个标记上一个元素找到没，一个是下一个元素是否重复查到过。
                sta.push(index);
                k++;//字符窜str下标推进
                if (k == str.length) return true;//已经遍历完str
            } else {
                sta.pop();
                k--;
            }
            back[index] = 1;//无论有无都标记
        }
        return false;
    }
}
public class Main {
    public static void main(String[] args) {
        char[] temp = {'a','b','c','e','s','f','c','s','a','d','e','e'};
        char[] str = {'a','b','f','s'};
        Solution p = new Solution();
        boolean b = p.hasPath(temp,3,4,str);
        System.out.println("输出结果"+ b);
    }
}