《剑指offer》12--矩阵中的路径[C++]

题目描述

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

解题思路

使用回溯法（backtracking）进行求解，它是一种暴力搜索方法，通过搜索所有可能的结果来求解问题。回溯法在一次搜索结束时需要进行回溯（回退），将这一次搜索过程中设置的状态进行清除，从而开始一次新的搜索过程。例如下图示例中，从 f 开始，下一步有 4 种搜索可能，如果先搜索 b，需要将 b 标记为已经使用，防止重复使用。在这一次搜索结束之后，需要将 b 的已经使用状态清除，并搜索 c。

 

递归DFS

class Solution {
public:
     bool hasPath(char* matrix, int rows, int cols, char* str) {
        if(str==NULL||rows<=0||cols<=0) return false;
        vector<bool> isOk (rows*cols, false);
        for(int i=0; i<rows; i++)
            for(int j=0; j<cols; j++)
                if(isHsaPath(matrix,rows,cols,str,isOk,i,j)) return true;
        return false;
    }
     bool isHsaPath(char *matrix, int rows, int cols, char *str, vector<bool> isOk, int curx, int cury) {
          if(*str == '\0') return true;
          int index = curx*cols+cury;
          if(curx<0 || curx>=rows || cury<0 || cury>=cols || isOk[index] || *str!=matrix[index]) return false;
          isOk[index] = true;//要走的第一个位置置为true，表示已经走过了
          if(isHsaPath(matrix,rows,cols,str+1,isOk,curx-1,cury)
           || isHsaPath(matrix,rows,cols,str+1,isOk,curx+1,cury)
           || isHsaPath(matrix,rows,cols,str+1,isOk,curx,cury-1)
           || isHsaPath(matrix,rows,cols,str+1,isOk,curx,cury+1)) return true;
          isOk[index] = false;//走到这，说明这一条路不通，还原，再试其他的路径
          return false;
     }
};

非递归标准DFS

class Solution {
public:
    typedef pair<int, int> position;
    struct state {
        position p;
        int s;
        vector<bool> v;
        state(position pos, int step, vector<bool> visited):
            p(pos), s(step), v(visited) {}
    };
    int dx[4] = {0, 0, -1, 1}, dy[4] = {-1, 1, 0, 0};
    bool hasPath(char* matrix, int rows, int cols, char* str) {
        if(str==NULL||rows<=0||cols<=0) return false;
        stack<state> q;
        int maxS = strlen(str)-1;
        vector<bool> v(rows*cols, false);
        for(int x=0; x<rows; ++x) {
            for(int y=0; y<cols; ++y) {
                if(matrix[x*cols+y] == str[0]) {
                    v[x*cols+y] = true;
                    q.push(state(position(x,y), 0, v));
                    v[x*cols+y] = false;
                }
            }
        }
        while(!q.empty()) {
            auto t = q.top(); q.pop();
            auto p = t.p; auto x = p.first; auto y = p.second;
            auto s = t.s;
            auto v = t.v;
            if(s == maxS) return true;
            for(int d=0; d<4; ++d) {
                int nx = x+dx[d], ny = y+dy[d], ns = s+1;
                if((nx>=0 && nx<rows) && (ny>=0 && ny<cols) && ns<=maxS 
                   && !v[nx*cols+ny] && matrix[nx*cols+ny]==str[ns]) {
                    v[nx*cols+ny] = true;
                    q.push(state(position(nx, ny), ns, v));
                    //v[nx*cols+ny] = false;//可以注释掉，为了安全，建议保留
                }
            }
        }
        return false;
    }
};

参考文献

[1] CS-Notes