矩阵中的路径

矩阵中的路径

http://www.nowcoder.com/questionTerminal/c61c6999eecb4b8f88a98f66b273a3cc

回溯法:

//
// Created by jt on 2020/9/18.
//
#include <vector>
using namespace std;

class Solution {
public:
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        vector<vector<int> > visited(rows, vector<int>(cols, 0));
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if (backtrace(matrix, rows, cols, str, visited, i, j, 0)) return true;
            }
        }
        return false;
    }

    bool backtrace(char *matrix, int rows, int cols, char *str, vector<vector<int> > &visited, int x, int y, int pos) {
        if (x < 0 || x >= rows || y < 0 || y >= cols) return false;
        if (visited[x][y] || matrix[x*cols + y] != str[pos]) return false;
        if (pos == strlen(str) - 1) return true;
        visited[x][y] = 1;
        // 上
        bool a = backtrace(matrix, rows, cols, str, visited, x - 1, y, pos+1);
        // 下
        bool b = backtrace(matrix, rows, cols, str, visited, x + 1, y, pos+1);
        // 左
        bool c = backtrace(matrix, rows, cols, str, visited, x, y - 1, pos+1);
        // 右
        bool d = backtrace(matrix, rows, cols, str, visited, x, y + 1, pos+1);
        visited[x][y] = 0;
        return a || b || c || d;
    }
};
刷遍天下无敌手 文章被收录于专栏

秋招刷题历程

全部评论

相关推荐

10-30 22:18
已编辑
毛坦厂中学 C++
点赞 评论 收藏
分享
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-24 20:55
阿里国际 Java工程师 2.7k*16.0
程序员猪皮:没有超过3k的,不太好选。春招再看看
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务