【 LeetCode】剑指 Offer 12. 矩阵中的路径 (中等) dfs回溯

剑指 Offer 12. 矩阵中的路径
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。

[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]

但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。

示例 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

代码:

class Solution {
    
    
public:
    bool vis[220][220]={
    
    0}; //标记是否走过
    int dx[4]={
    
    0,0,-1,1},flag=0; //四个方向,flag为1代表找到路径
    int dy[4]={
    
    1,-1,0,0};

    void dfs(int x,int y,int index,int len1,int len2,vector<vector<char>>& board, string word)
    {
    
    
        if(flag) return; //剪枝

        if(index==word.size()) //找完该路径
        {
    
    
            flag=1; //标记
            return;
        }

        for(int i=0;i<4;i++)
        {
    
    
            int newx=x+dx[i]; //下个位置
            int newy=y+dy[i];

            if(newx>=0 && newx<len1 && newy>=0 && newy<len2 && !vis[newx][newy]) //在边界内 && 没访问过
            {
    
    
                if(word[index]!=board[newx][newy]) continue; //不匹配,跳过
                vis[newx][newy]=1;
                dfs(newx,newy,index+1,len1,len2,board,word);
                vis[newx][newy]=0; //回溯
            }
        }
    }

    bool exist(vector<vector<char>>& board, string word) {
    
    
        int len1=board.size(),len2=board[0].size();
        for(int i=0;i<len1;i++)
        {
    
    
            for(int j=0;j<len2;j++)
            {
    
    
                if(board[i][j]==word[0])
                {
    
    
                    vis[i][j]=1;
                    dfs(i,j,1,len1,len2,board,word);
                    vis[i][j]=0;
                    if(flag) return 1; //找到路径,返回
                }
            }
        }
        return 0; //没找到
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_45260385/article/details/108453106