牛客网——矩阵中的路径（C++）

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

C++

class Solution {
public:
    void dfs(char* matrix, int rows, int cols, char* str, vector<vector<int>>& flag, int k, int r, int c, int& tmp)
    {
        int len=strlen(str);
        int dx[]={1,-1,0,0};
        int dy[]={0,0,1,-1};
        int start=r*cols+c;
        if(0==flag[r][c])
        {
            if(matrix[start]==str[k])
            {
                if(k==len-1)
                {
                    tmp=1;
                    return;
                }
                flag[r][c]=1;
                for(int s=0;s<4;s++)
                {
                    int y=r+dy[s];
                    int x=c+dx[s];
                    if(y>=0 && y<rows && x>=0 && x<cols)
                    {
                        dfs(matrix,rows,cols,str,flag,k+1,y,x,tmp);
                    }
                }
                flag[r][c]=0;  
                return; 
            }
            else
            {
                return;
            }
        }
    }

    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        int n=strlen(str);
        if(0==n)
        {
            return true;
        }
        vector<vector<int>> flag(rows,vector<int>(cols,0));
        int tmp=0;
        for(int i=0;i<rows;i++)
        {
            for(int j=0;j<cols;j++)
            {
                dfs(matrix,rows,cols,str,flag,0,i,j,tmp);
                if(tmp)
                {
                    return true;
                }
            }
        }
        return false;
    }


};