剑指 Offer(第2版)面试题 12:矩阵中的路径
- 剑指 Offer(第2版)面试题 12:矩阵中的路径
- 解法1:回溯
剑指 Offer(第2版)面试题 12:矩阵中的路径
题目来源:23. 矩阵中的路径
解法1:回溯
回溯算法模板题。
我们先枚举单词的起点,然后依次枚举单词的每个字母。
过程中需要将已经使用过的字母改成一个特殊字母(‘*’),以避免重复使用字符,注意回溯是要把特殊字母改回原来的状态,这也是回溯算法的精髓。
代码:
class Solution
{
private:const int dx[4] = {-1, 0, 1, 0};const int dy[4] = {0, 1, 0, -1};public:bool hasPath(vector<vector<char>> &matrix, string &str){// 特判if (matrix.empty())return false;int m = matrix.size(), n = m ? matrix[0].size() : 0;for (int i = 0; i < m; i++)for (int j = 0; j < n; j++)if (backtracking(matrix, str, 0, i, j))return true;return false;}// 辅函数 - 回溯bool backtracking(vector<vector<char>> &matrix, string &str, int level, int x, int y){if (matrix[x][y] != str[level])return false;if (level == str.size() - 1)return true;char c = matrix[x][y];matrix[x][y] = '*';for (int i = 0; i < 4; i++){int r = x + dx[i], c = y + dy[i];if (r >= 0 && r < matrix.size() && c >= 0 && c < matrix[0].size())if (backtracking(matrix, str, level + 1, r, c))return true;}matrix[x][y] = c;return false;}
};
复杂度分析:
时间复杂度:O(m*n*3k),其中 m 和 n 分别是二维矩阵 matrix 的行数和列数。遍历二维矩阵 matrix 的每个元素,作为单词的起点,单词的每个字母一共有上下左右四个方向可以选择,但由于不能走回头路,所以除了单词首字母外,仅有 3 种选择。
空间复杂度:O(1)。