劍指offer 66. 機器人的運動範圍
阿新 • • 發佈:2018-11-29
題目描述
地上有一個m行和n列的方格。一個機器人從座標0,0的格子開始移動,每一次只能向左,右,上,下四個方向移動一格,但是不能進入行座標和列座標的數位之和大於k的格子。 例如,當k為18時,機器人能夠進入方格(35,37),因為3+5+3+7 = 18。但是,它不能進入方格(35,38),因為3+5+3+8 = 19。請問該機器人能夠達到多少個格子?
思路:
同上一題思路一樣,判斷條件改成了行座標和列座標的數位之和大於k。
此外,這裡我可以不再採用mark_matrix,而使用一個list進行座標標記self.index_list.append([x, y])
。
參考答案:
# -*- coding:utf-8 -*-
class Solution:
def movingCount(self, threshold, rows, cols):
# write code here
if rows < 0 or cols <0:
return 0
self.count = 0
self.index_list = []
self.destGrid(threshold, 0, 0, rows, cols)
return self.count
def destGrid(self, threshold, row, col, rows, cols):
if row >= rows or col >= cols:
return
for x in range(row, rows):
for y in range(col, cols):
if self.index_sum(x, y) <= threshold and [x, y] not in self.index_list:
self. count += 1
self.index_list.append([x, y])
self.destGrid(threshold, x+1, y, rows, cols)
self.destGrid(threshold, x, y+1, rows, cols)
else:
return
def index_sum(self, row_index, col_index):
index_sum = 0
while row_index > 0:
index_sum += row_index % 10
row_index = int(row_index / 10)
while col_index > 0:
index_sum += col_index % 10
col_index = int(col_index / 10)
return index_sum
Note
- 這種找路徑尤其表明不能走回頭路的題,往往需要結合一個標記矩陣用於標識該矩陣對應元素是否為已經遍歷過了,如果變數過了,則不能再進行重複遍歷。