剑指offer66 机器人的运动范围

最新推荐文章于 2024-09-18 18:23:55 发布

yanglee0

最新推荐文章于 2024-09-18 18:23:55 发布

阅读量62

点赞数

分类专栏：剑指offer 文章标签：算法

本文链接：https://blog.csdn.net/qq_38147421/article/details/109742166

版权

剑指offer 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

机器人的运动范围

在这里插入图片描述
思路：将地图全部置1，遍历能够到达的点，将遍历的点置0并令计数+1.这个思路在找前后左右相连的点很有用，比如leetcode中的海岛个数问题/最大海岛问题都可以用这种方法来求解。

# -*- coding:utf-8 -*-
class Solution:
    count = 0
    def movingCount(self, threshold, rows, cols):
        # write code here
        arr = [[1 for i in range(cols)] for j in range(rows)]
        self.findway(arr, 0, 0, threshold)
        return self.count
    def findway(self,arr,i,j,k):
        if i < 0 or j < 0 or i >= len(arr) or j >= len(arr[0]):
            return
        tempi = list(map(int,list(str(i))))
        tempj = list(map(int,list(str(j))))
        if sum(tempi)+sum(tempj)>k or arr[i][j]!=1:
            return
        arr[i][j] = 0
        self.count +=1
        self.findway(arr, i+1, j, k)
        self.findway(arr, i-1, j, k)
        self.findway(arr, i, j+1, k)
        self.findway(arr, i, j-1, k)