剑指offer :矩阵中的路径

题目:

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

思路:深度优先加回朔

class Solution {
public:
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        if (NULL == str ||0 == strlen(str))
            return true;
        int str_len = strlen(str);
        vector<vector<bool> >flag(rows, vector<bool> (cols, false));
        
        for (int i=0; i<rows; i++)
        {
            for (int j=0; j<cols; j++)
            {
                if(matrix[i*cols+j] == str[0])
                {
                    flag[i][j] = true;
                    if (DFS(matrix, rows, cols, i, j, str, str_len, 1, flag))
                        return true;
                    //这里一定要复位
                    flag[i][j] = false;
                }
            }
        }
        return false;
    }
    
    bool DFS(char* matrix, int rows, int cols, int cur_rows, int cur_cols, char* str, int str_len, int str_cur_len, vector<vector<bool> > &flag)
    {
        if (str_cur_len == str_len)
            return true;
        if (cur_rows < rows-1 && !flag[cur_rows + 1][cur_cols] && str[str_cur_len] == matrix[(cur_rows + 1)*cols+cur_cols])
        {
            flag[cur_rows + 1][cur_cols] = true;
            if (DFS(matrix, rows, cols, cur_rows + 1, cur_cols, str, str_len, str_cur_len+1, flag))
                return true;
            //这里一定要复位
            flag[cur_rows + 1][cur_cols] = false;
        }
        
        if (cur_rows > 0 && !flag[cur_rows - 1][cur_cols] && str[str_cur_len] == matrix[(cur_rows - 1)*cols+cur_cols])
        {
            flag[cur_rows - 1][cur_cols] = true;
            if (DFS(matrix, rows, cols, cur_rows - 1, cur_cols, str, str_len, str_cur_len+1, flag))
                return true;
            flag[cur_rows - 1][cur_cols] = false;
        }
        if (cur_cols > 0 && !flag[cur_rows][cur_cols - 1] && str[str_cur_len] == matrix[cur_rows*cols + cur_cols - 1])
        {
            flag[cur_rows][cur_cols - 1] = true;
            if (DFS(matrix, rows, cols, cur_rows, cur_cols - 1, str, str_len, str_cur_len+1, flag))
                return true;
            flag[cur_rows][cur_cols - 1] = false;
        }
        if (cur_cols < cols-1 && !flag[cur_rows][cur_cols + 1] && str[str_cur_len] == matrix[cur_rows*cols + cur_cols + 1])
        {
            flag[cur_rows][cur_cols + 1] = true;
            if (DFS(matrix, rows, cols, cur_rows, cur_cols + 1, str, str_len, str_cur_len+1, flag))
                return true;
            flag[cur_rows][cur_cols + 1] = false;
        }
        return false;
    }

};

猜你喜欢

转载自blog.csdn.net/pynash123/article/details/89069703