【力扣100】【好题】79.单词搜索

添加链接描述

class Solution(object):
    
    # 定义上下左右四个行走方向
    directs = [(0, 1), (0, -1), (1, 0), (-1, 0)]
    
    def exist(self, board, word):
        """
        :type board: List[List[str]]
        :type word: str
        :rtype: bool
        """
        m = len(board)
        if m == 0:
            return False
        n = len(board[0])
        mark = [[0 for _ in range(n)] for _ in range(m)]
                
        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] == word[0]:
                    # 将该元素标记为已使用
                    mark[i][j] = 1
                    if self.backtrack(i, j, mark, board, word[1:]) == True:
                        return True
                    else:
                        # 回溯
                        mark[i][j] = 0
        return False
        
        
    def backtrack(self, i, j, mark, board, word):
        if len(word) == 0:
            return True
        
        for direct in self.directs:
            cur_i = i + direct[0]
            cur_j = j + direct[1]
            
            if cur_i >= 0 and cur_i < len(board) and cur_j >= 0 and cur_j < len(board[0]) and board[cur_i][cur_j] == word[0]:
                # 如果是已经使用过的元素,忽略
                if mark[cur_i][cur_j] == 1:
                    continue
                # 将该元素标记为已使用
                mark[cur_i][cur_j] = 1
                if self.backtrack(cur_i, cur_j, mark, board, word[1:]) == True:
                    return True
                else:
                    # 回溯
                    mark[cur_i][cur_j] = 0
        return False

思路:

  1. 首先在board中找到target中的第一个字母
  2. 为什么要使用一个mark数组记录是否使用过?为了应对下面这个情况:
    【力扣100】【好题】79.单词搜索_第1张图片
  3. 为什么要回溯?也是上面这个情况,比如我一个s周围有4个e,第一个e用不了,那就回尝试第二个,第三个,第四个…所以需要回溯
  4. directs数组用来进行扫荡,而且所有循环都是在这个扫荡的基础上进行的
  5. 边界值条件,不能比0小,也不能超过最大值,并且需要字母匹配
  6. 很全面的一道题,值得反复做

你可能感兴趣的:(leetcode,算法)