剑指offer：矩阵中的路径

题目描述

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

思路

//回溯法
//用一个标记数组visited记录是否走过这个格子，从每个满足字符串第一个字符的点开始走，判断上下左右四个格子是不是在字符串下一位，走没走过，满足不满足K的要求
//满足的话就走这个结点，设置visited=true，字符串指针后移。

code

class Solution {
public:
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        if(matrix==nullptr||rows<1||cols<1||str==nullptr)
            return false;
        bool *visited=new bool[rows*cols];
        memset(visited,0,rows*cols);
        int length=0;
        for(int row=0;row<rows;++row){
            for(int col=0;col<cols;++col){
                if(hasPathCore(matrix,rows,cols,row,col,str,visited,length))
                    return true;
            }
        }
        delete []visited;
        return false;
    }
    bool hasPathCore(const char*matrix,const int&rows,const int &cols,int row, int col,char*&str,bool*&visited,int &length){
        if(str[length]=='\0')
            return true;
        bool hasPath=false;
        if(col>=0&&col<cols&&row>=0&&row<rows&&matrix[row*cols+col]==str[length]&&!visited[row*cols+col]){
            ++length;
            visited[row*cols+col]=true;
            hasPath=hasPathCore(matrix,rows,cols,row-1,col,str,visited,length)||
                    hasPathCore(matrix,rows,cols,row+1,col,str,visited,length)||
                    hasPathCore(matrix,rows,cols,row,col-1,str,visited,length)||
                    hasPathCore(matrix,rows,cols,row,col+1,str,visited,length);
            if(!hasPath){
                length--;
                visited[cols*row+col]=false;
            }
        }
        return hasPath;
    }
};