题目描述:
地上有一个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。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:
输入:m = 3, n = 1, k = 0
输出:1
提示:
1 <= n,m <= 100
0 <= k <= 20
解题思路:
一、深度优先
1.深度优先,递归调用,当一个坐标满足条件时,继续搜索其相邻的格子
2.将访问过的格子进行标注
3.由于格子都是相连的,所以在搜索时,只向右下搜索即可
# def getSum(x):
# count=0
# while x!=0:
# count=x%10
# x=x//10
# return count
class Solution:
def movingCount(self, m: int, n: int, k: int) -> int:
def search(i,j):
if i<0 or i>=m or j<0 or j>=n or (i//10 + i%10 + j//10 + j%10)>k or visit[i][j]==1:
return 0
visit[i][j]=1
return 1+search(i+1,j)+search(i-1,j)+search(i,j+1)+search(i,j-1)
visit=[[0]*n for i in range(m)]
#print(visit)
return search(0,0)
二、广度优先(队列)
1.队列queue控制接下来判断的节点,满足的格子放在set中
2。将当前格子的右下格子放在队列中,每次从队列前端取出数据进行判断
def digitsum(n):
ans = 0
while n:
ans += n % 10
n //= 10
return ans
class Solution:
def movingCount(self, m: int, n: int, k: int) -> int:
from queue import Queue
q = Queue()
q.put((0, 0))
s = set()
while not q.empty():
x, y = q.get()
if (x, y) not in s and 0 <= x < m and 0 <= y < n and digitsum(x) + digitsum(y) <= k:
s.add((x, y))
for nx, ny in [(x + 1, y), (x, y + 1)]:
q.put((nx, ny))
return len(s)