剑指 Offer 12. 矩阵中的路径
了解到整个算法流程,需要动手去尝试,看看递归的入口,减枝的位置,整个回溯函数的结构。
一次通过,虽然也不确定自己逻辑是否正确,所以需要自己去尝试
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
def contains(x, y, key):
if key == self.length: return True
if x<0 or y<0 or x>=self.row or y>=self.col:
return False
if self.board[x][y] == self.word[key]:
self.board[x][y] = ' '
res = contains(x+1, y, key+1) or contains(x-1, y, key+1) or contains(x, y+1, key+1) \
or contains(x, y-1, key+1)
self.board[x][y] = self.word[key]
return res
else:
return False
self.row, self.col = len(board), len(board[0])
self.length = len(word)
self.board = board
self.word = word
for x in range(self.row):
for y in range(self.col):
res = contains(x, y, 0)
if res: return True
return False
这里作者k=word.length - 1就代表匹配完成的原因是:
他先判断这一步的单词是否一样,再判断k=word.length - 1。
而我是先判断k=word.length,在做后续
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
def dfs(i, j, k):
if not 0 <= i < len(board) or not 0 <= j < len(board[0]) or board[i][j] != word[k]: return False
if k == len(word) - 1: return True
board[i][j] = ''
res = dfs(i + 1, j, k + 1) or dfs(i - 1, j, k + 1) or dfs(i, j + 1, k + 1) or dfs(i, j - 1, k + 1)
board[i][j] = word[k]
return res
for i in range(len(board)):
for j in range(len(board[0])):
if dfs(i, j, 0): return True
return False
剑指 Offer 13. 机器人的运动范围
地上有一个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。请问该机器人能够到达多少个格子?
有些方格满足条件,但是机器人无法移到到那里
方法一:深度优先遍历 DFS
方法二:广度优先遍历 BFS
python集合Set()
python列表实现队列,list.pop(0) and list.append()
DFS
BFS
def sums(x):
s = 0
while x != 0:
s += x % 10
x = x // 10
return s
class Solution:
def movingCount(self, m: int, n: int, k: int) -> int:
def dfs(i, j):
key = sums(i) + sums(j)
# 只需向下和向右走
if i >= m or j >= n or k < key or (i, j) in visited: return 0
# 当前位置成立
visited.add((i,j))
return 1 + dfs(i + 1, j) + dfs(i, j + 1)
visited = set()
return dfs(0, 0)
from queue import Queue
class Solution:
def movingCount(self, m: int, n: int, k: int) -> int:
queue, visited = [(0, 0)], set()
# queue, visited = Queue(), set()
# queue.put((0,0))
while queue:
# while not queue.empty():
i, j = queue.pop(0) # list.pop([index=-1])
# i, j = queue.get()
key = sums(i) + sums(j)
if i >= m or j >= n or k < key or (i, j) in visited: continue
# 当前位置成立
visited.add((i,j))
queue.append((i + 1, j))
queue.append((i, j + 1))
# queue.put((i + 1, j))
# queue.put((i, j + 1))
return len(visited)