剑指 Offer 12. 矩阵中的路径

题目描述

思路分析

先在原矩阵中寻找到字符串的第一个字母，然后再对这个字母周围矩阵进行深度优先搜索。上下左右开始搜索，改变搜索路径的字符，搜索完成后还原，防止重复搜索。

代码展示

class Solution {
public:
    bool exist(vector<vector<char>>& board, string word) {
        for (int row = 0; row < board.size(); row++) {
            for (int col = 0; col < board[0].size(); col++) {
                if (board[row][col] == word[0]) {
                    if (dfs(board, word, row, col, 0) == true) {
                        return true;
                    }
                }
            }
        }
        return false;
    }
    bool dfs(vector<vector<char>>& board, string& word, int row, int col, int pos) {
        if (row < 0 || row > board.size() -1 || col < 0 || col > board[0].size() - 1 || board[row][col] != word[pos]) {
            return false;
        }
        if (pos == word.size() -1) {
            return true;
        }
        char tmp = board[row][col];
        board[row][col] = '#';
        bool m = dfs(board, word, row+1, col, pos+1)||dfs(board, word, row-1, col, pos+1)||dfs(board, word, row, col+1, pos+1)||dfs(board, word, row, col-1, pos+1);
        board[row][col] = tmp;
        return m;
    }
};

结果分析

M,N 分别为矩阵行列大小， KK 为字符串 word 长度。

时间复杂度 O(3^KMN) ： 最差情况下，需要遍历矩阵中长度为 KK 字符串的所有方案，时间复杂度O(3^K)
矩阵中共有 MN 个起点，时间复杂度为 O(MN)
空间复杂度O(K)
空间上只用了调用函数的函数栈