【leetcode】机器人的运动范围（递归回溯）

最新推荐文章于 2024-10-11 11:20:04 发布

hustfc

最新推荐文章于 2024-10-11 11:20:04 发布

阅读量1k

点赞数

分类专栏： python 基础算法

本文链接：https://blog.csdn.net/weixin_36372879/article/details/83185212

版权

python 同时被 2 个专栏收录

141 篇文章 2 订阅

订阅专栏

基础算法

138 篇文章 1 订阅

订阅专栏

本文详细解析LeetCode中关于机器人运动范围的问题，通过递归回溯的方法来求解。内容包括问题描述、解题思路、代码实现及测试用例分析，帮助读者掌握这种经典的搜索算法。

摘要由CSDN通过智能技术生成

在这里插入图片描述

# -*- coding:utf-8 -*-
class Solution:
    def getDigitSum(self, num):
        sum = 0
        while num > 0:
            sum += num % 10
            num = int(num / 10)
        return sum
    def movingCore(self, threshold, rows, cols, row, col, visited):
        count = 0
        if row >= 0 and row < rows and col >= 0 and col < cols and not visited[row][col] \
            and self.getDigitSum(row) + self.getDigitSum(col) <= threshold:
            visited[row][col] = 1
            count = 1 + self.movingCore(threshold, rows, cols, row + 1, col, visited) \
            +self.movingCore(threshold, rows, cols, row - 1, col, visited) \
            +self.movingCore(threshold, rows, cols, row, col + 1, visited) \
            +self.movingCore(threshold, rows, cols, row, col - 1, visited)
        return count
    def movingCount(self, threshold, rows, cols):
        if threshold < 0 or rows <= 0 or cols <= 0:
            raise Exception('Input Error')
        visited = [([0] * cols) for i in range(rows)]
        count = self.movingCore(threshold, rows, cols, 0, 0, visited)
        return count
rows = cols = 20
print(Solution().movingCount(18, rows, cols))