(python)小菜狗算法日记(递归DFS系列)_leetcode 面试题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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ju-zhen-zhong-de-lu-jing-lcof

提示:

  • 1 <= board.length <= 200
  • 1 <= board[i].length <= 200

DFS递归,一般找路径用DFS,找最短之类的用BFS

先写row,col是因为DFS里面要用到,保证上下左右移动后的ni,nj不越界

递归终止条件是吧word字符串中所有的字符都走完,cur是从第二个字符,index=1进入递归,则当cur指向字符串最后一个字符时,再+1进入下一层递归,但此时cur指针不指向任何字符串,因为已经指完了,此时cur==len(word),直接return,回到指向最后字符的那层递归。

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        row = len(board)
        col = len(board[0])
        def DFS(i,j,cur,visited):
            if cur == len(word):
                return True
            pos = ((0,-1),(-1,0),(1,0),(0,1))
            for m,n in pos:
                ni,nj = i+m,j+n
                # 满足不越界,且坐标没访问过且坐标值等于字符串中的下一字符,继续递归
                if 0<=ni<row and 0<=nj<col and (ni,nj) not in visited and board[ni][nj]==word[cur]:
                    # 这里如果不加这个判断返回true,则会返回for循环外的False
                    if DFS(ni,nj,cur+1,visited+[(ni,nj)]):
                        return True
            #如果返回False,返回上一层递归看有没有其他路径满足,如果都没有就出递归返回False
            return False
        
        # DFS如果返回False就再找其他入口,其他入口也不行就返回False
        for i in range(row):
            for j in range(col):
                if board[i][j]==word[0]: #找到入口
                    if DFS(i,j,1,[(i,j)]):
                        return True
        return False

牛客上这个矩阵的路径题的输入给的不是矩阵是个字符串,得自己加一句:

matrix = [list(matrix[cols*i:cols*(i+1)]) for i in range(rows)]

牛客版本:

# -*- coding:utf-8 -*-
class Solution:
    def hasPath(self, matrix, rows, cols, path):
        # write code here
 
        def DFS(i,j,cur,visited):
            if cur == len(path):
                return True
            pos = ((0,-1),(-1,0),(1,0),(0,1))
            for m,n in pos:
                ni,nj = i+m,j+n
                # 满足不越界,且坐标没访问过且坐标值等于字符串中的下一字符,继续递归
                if 0<=ni<rows and 0<=nj<cols and (ni,nj) not in visited and matrix[ni][nj]==path[cur]:
                    # 这里如果不加这个判断返回true,则会返回for循环外的False
                    if DFS(ni,nj,cur+1,visited+[(ni,nj)]):
                        return True
            #如果返回False,返回上一层递归看有没有其他路径满足,如果都没有就出递归返回False
            return False
         
        # DFS如果返回False就再找其他入口,其他入口也不行就返回False
        matrix = [list(matrix[cols*i:cols*i+cols]) for i in range(rows)]
        for i in range(rows):
            for j in range(cols):
                if matrix[i][j]==path[0]: #找到入口
                    if DFS(i,j,1,[(i,j)]):
                        return True
        return False

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值