剑指 Offer 12. 矩阵中的路径

思路

dfs +剪枝

 

代码实现

class Solution {
private:
    int d[4][2] = {{0,1},{0,-1},{1,0},{-1,0}};
    int m, n;   //n行m列
    bool vis[210][210] = {false};
    
public:
    bool exist(vector<vector<char>>& board, string word) {
        n = board.size();
        m = board[0].size();
        
        for(int i = 0; i < n; ++i) {
            for(int j = 0; j < m; ++j) {
                //memset(vis, false, sizeof(vis));
                //vector<vector<bool>> vis(n, vector<bool>(m, false)); //使用vector会超时，不知道为啥
                if(dfs(board, word, i, j, 0))
                    return true;
            }
        }
        return false;
    }

    //当前搜索第index个字符
    bool dfs(vector<vector<char>>& board, string targetWord, int x, int y, int index) {
        
        //剪枝
        if(board[x][y] != targetWord[index])
            return false;
        
        //剪枝
        if(index == targetWord.length()-1)
            return true;

        vis[x][y] = true;

        for(int i = 0; i < 4; ++i) {
            int xx = x + d[i][0];
            int yy = y + d[i][1];
            if(xx < 0 || yy < 0 || xx >= n || yy >= m || vis[xx][yy]) 
                continue;
　　　　　　　//找到一个答案递归即可返回，不用继续往后找其他答案
            if(dfs(board, targetWord, xx, yy, index+1))
                return true;
            
        }

        vis[x][y] = false;  //记得回溯
        return false;
    }
};

 

复杂度分析

 

 

参考

面试题12. 矩阵中的路径（深度优先搜索 DFS ，清晰图解）