【剑指Offer】搜索与回溯算法(2)

最新推荐文章于 2024-09-21 14:48:47 发布

Ryu眼神不太好

最新推荐文章于 2024-09-21 14:48:47 发布

阅读量151

点赞数 1

分类专栏：刷题文章标签：算法数据结构

本文链接：https://blog.csdn.net/Chris_Eddy/article/details/120425500

版权

刷题专栏收录该内容

36 篇文章 0 订阅

订阅专栏

剑指 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 仅由大小写英文字母组成

class Solution {
public:
    vector<vector<int>> dirs = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
    bool dfs(int x, int y, vector<vector<char>> &board, const string &word, int pos){
        if(pos == word.size())return true;
        int n = board.size(), m = board[0].size();
        char ch = board[x][y];
        board[x][y] = '#';
        for(int i = 0; i < 4; ++ i){
            int nx = x + dirs[i][0], ny = y + dirs[i][1];
            if(nx >= 0 && nx < n && ny >= 0 && ny < m){
                if(board[nx][ny] == word[pos]){
                    if(dfs(nx, ny, board, word, pos + 1) == true){
                        board[x][y] = ch;
                        return true;
                    }
                }
            }
        }
        board[x][y] = ch;
        return false;
    }
    bool exist(vector<vector<char>>& board, const string &word) {
        int n = board.size(), m = board[0].size();
        for(int i = 0; i < n; ++ i){
            for(int j = 0; j < m; ++ j){
                if(board[i][j] == word[0]){
                    if(dfs(i, j, board, word, 1) == true){
                        return true;
                    }
                }
            }
        }
        return false;
    }
};

算法思路

很常规的深度优先搜索，每一步判断字符是否正确，然后往四个方向前进，遍历过的有用字符设为’#’，这样就避免重复访问，当访问到字符串最后一个字符串时则说明矩阵中存在字符串，返回true，否则返回false。

题目一结果

地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0]的格子开始移动，它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。例如，当k为18时，机器人能够进入方格 [35, 37] ，因为3+5+3+7=18。但它不能进入方格 [35, 38]，因为3+5+3+8=19。请问该机器人能够到达多少个格子？

示例 1：

输入：m = 2, n = 3, k = 1
输出：3

示例 2：

输入：m = 3, n = 1, k = 0
输出：1

提示：

1 <= n,m <= 100
0 <= k <= 20

class Solution {
public:
    vector<vector<bool>> vis;
    vector<vector<int>> dirs = {{0, 1}, {1, 0}, {-1, 0}, {0, -1}};
    bool judge(int x, int y, int k){
        while(x){
            k -= (x % 10);
            x /= 10;
        }
        while(y){
            k -= (y % 10);
            y /= 10;
        }
        return k >= 0;
    }
    void dfs(int x, int y, int n, int m, int k){
        vis[x][y] = true;
        for(int i = 0; i < 4; ++ i){
            int nx = x + dirs[i][0], ny = y + dirs[i][1];
            if(nx >= 0 && nx < n && ny >= 0 && ny < m){
                if(vis[nx][ny] == false && judge(nx, ny, k) == true){
                    dfs(nx, ny, n, m, k);
                }
            }
        }
    }
    int movingCount(int n, int m, int k) {
        int ans = 0;
        vis.resize(n, vector<bool>(m));
        dfs(0, 0, n, m, k);
        for(int i = 0; i < n; ++ i){
            for(int j = 0; j < m; ++ j){
                if(vis[i][j])ans ++;
            }
        }
        return ans;
    }
};

算法思路

这题可以用深度优先搜索和广度优先搜索，我用的是广度优先搜索，和第一题差不多，每走一步就判断是否合法，合法就继续遍历，不合法就返回，注意题目求的是符合条件的个数，因此vis矩阵不用重置，最后统计总共访问过的格子个数就是答案。

题目二结果