剑指offer——矩阵中的路径（c++)

题目描述

设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。
例如，下面的3*4矩阵
[
[a b t g]
[c f c s]
[j d e h]
]
包含字符串"bfce"的路径

思路

首先在矩阵中任意选取一个格子作为起点。假设矩阵中某个格子的字符为ch，并且这个格子对应于路径上的第i个字符。如果路径上的第i个字符不是ch，那么这个格子不可能处在路径的第i个位置上。如果路径的第i个字符恰好是ch，那么到相邻的格子上寻找第i+1个字符。重复这个过程直到路径上的所有字符都在矩阵中找到相应的位置。
由于回溯法的递归特性，路径可以被看做一个栈。

class Solution {
public:
    bool exist(vector<vector<char>>& board, string word) {
        if (board.empty() || board[0].empty()) 
            return false;
        int m = board.size(), n = board[0].size();
        vector<vector<bool>> visited(m, vector<bool>(n, false));
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                if (search(board, word, 0, i, j, visited)) 
                    return true;
            }
        }
        return false;
    }
    bool search(vector<vector<char>>& board, string word, int idx, int i, int j, vector<vector<bool>>& visited) {
        if (idx == word.size())
            return true;
        int m = board.size(), n = board[0].size();
        if (i < 0 || j < 0 || i >= m || j >= n || visited[i][j] || board[i][j] != word[idx]) 
            return false;
        visited[i][j] = true;
        bool res = search(board, word, idx + 1, i - 1, j, visited) || 
                   search(board, word, idx + 1, i + 1, j, visited) || 
                   search(board, word, idx + 1, i, j - 1, visited) || 
                   search(board, word, idx + 1, i, j + 1, visited);
        visited[i][j] = false;
        return res;
    }
};