剑指 Offer 12. 矩阵中的路径

题目描述

给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中，返回 true ；否则，返回 false 。

单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

例如，在下面的 3×4 的矩阵中包含单词 “ABCCED”（单词中的字母已标出）。

示例 1：

输入：board = [[“A”,“B”,“C”,“E”],[“S”,“F”,“C”,“S”],[“A”,“D”,“E”,“E”]], word = “ABCCED”
输出：true
示例 2：

输入：board = [[“a”,“b”],[“c”,“d”]], word = “abcd”
输出：false

提示：

1 <= board.length <= 200
1 <= board[i].length <= 200
board 和 word 仅由大小写英文字母组成

分析

这是一个深度优先搜索的问题，我们从第一个元素开始遍历，循环遍历整个矩阵，在遍历每一个元素的时候，判断这个元素是否等于字符串第一个字符，如果不相等，就返回，如果相等，标记该元素已经被使用过，然后再判断这个元素的四个方向是否等于字符串的下一个字符，依次类推，如果到匹配完成则说明找到了一条路径，否则就返回false。

代码

class Solution {
public:
    int n, m;
    bool exist(vector<vector<char>>& board, string word) {
        if (board.empty() || board[0].empty()) return false;
        n = board.size(), m = board[0].size();
        
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++)
                //分别以i，j为起点开始遍历
                if (dfs(board, i, j, word, 0))
                    return true;
        return false;
    }

    bool dfs(vector<vector<char>>& board, int x, int y, string& word, int k) {
        //如果出界或者该处字符与字符串不相等，返回false
        if (x < 0 || x >= n || y < 0 || y >= m || word[k] != board[x][y]) return false;
        //如果遍历到最后一个字符，还没有返回false，返回true
        if (k == word.size() - 1) return true;
        //记录当前字符值
        char tmp = board[x][y];
        //改变当前字符为26个字母之外的，保证不再用该字符
        board[x][y] = '.';
        //偏移量
        int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
        for (int i = 0; i < 4; i++) {
            int a = x + dx[i], b = y + dy[i];
            //分别遍历，如果找到了返回true
            if (dfs(board, a, b, word, k + 1)) return true;
        }
        //恢复现场
        board[x][y] = tmp;
        return false;
    }
};