矩阵中的路径（剪枝法）——剑指Offer

题目描述

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

矩阵中包含一条字符串"bcced"的路径，但是矩阵中不包含"abcb"路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

题目分析

思路很明确，使用DFS递归的回溯剪枝思想，即添加一些判断条件使得程序不再递归下去。首先对于matrix中的每一个都可能是起点，需要遍历。由题可知，只要找到一条路径，即可返回true，最后的0表示从str的第0个字符开始。对于从每一个点开始的子路径，因为使用递归，我们只需知道在这一步该怎么做即可，不用管之后该怎么做。同时找到一个递归的出口即可。设置visited[]数组实现剪枝操作，详情请看注释。

代码

C++代码如下：

class Solution {
public:
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        if(matrix==nullptr||rows<=0||cols<=0||str==nullptr)
            return false;
        bool* visited=new bool[rows*cols];//创建访问数组
        memset(visited,0,rows*cols);//初始化访问数组
        int pathlen=0;
        for(int i=0;i<rows;i++)
        {
            for(int j=0;j<cols;j++)
        {
        if(pathgo(matrix,rows,cols,i,j,str,pathlen,visited))
            return true;
        }
        }
        delete [] visited;
        return false;//无路径返回false
        
    }
    
    bool pathgo(char* matrix, int rows, int cols,int i,int j, char* str,int pathlen,bool* visited)
    {
        //走到字符串最后一位，返回true
        if(str[pathlen]=='\0')
            return true;
        bool pathway=false;
        //进行递归判断
        if(i>=0&&i<rows&&j>=0&&j<cols&&matrix[i*cols+j]==str[pathlen]&&!visited[i*cols+j])
        {
            pathlen++;
            visited[i*cols+j]=1;
            pathway = pathgo(matrix,rows,cols,i-1,j,str,pathlen,visited)
                ||pathgo(matrix,rows,cols,i+1,j,str,pathlen,visited)
                ||pathgo(matrix,rows,cols,i,j-1,str,pathlen,visited)
                ||pathgo(matrix,rows,cols,i,j+1,str,pathlen,visited);
            //不符合条件的进行剪枝处理
            if(!pathway)
            {
                pathlen--;
                visited[i*cols+j]=0;
            }
        }
        return pathway;
    }


};