微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

递归函数返回值而不分配额外空间?

如何解决递归函数返回值而不分配额外空间?

所以我一直在尝试优化这个名为 Frog Jump 的 LeetCode 问题。以下是问题的基本描述:

给定一个按升序排列的石头位置(以单位为单位)的列表,确定青蛙是否能够通过落在最后一块石头上来过河。最初,青蛙在第一块石头上,假设第一次跳跃必须是 1 个单位。

如果青蛙的最后一次跳跃是 k 个单位,那么它的下一次跳跃必须是 k - 1、k 或 k + 1 个单位。注意青蛙只能向前跳跃。

例如:[0,1,3,5,6,8,12,17]

总共有 8 个石头。第 0 个单位的第一块石头,第二块 第一个单位的石头,第三个单位的第三个石头,依此类推...... 第 17 单元的最后一块石头。

返回真。青蛙可以通过跳跃 1 个单位跳到最后一块石头 到第 2 个石头,然后 2 个单位到第 3 个石头,然后 2 个单位到 第 4 个石头,然后 3 个单位到第 6 个石头,4 个单位到第 7 个石头, 和第 8 个石头 5 个单位。

这是我的解决方案。我需要帮助的是如何在不分配额外的 res 数组的情况下输出相同的布尔值,该数组存储使用 DFS 的所有探索路径的逻辑 OR。

class Solution:
    def canCross(self,stones: List[int]) -> bool:
        
        if stones[1] != 1:
            return False
        
        res = []
        memo = {}
        
        def dfs(curr_stone,last_stone,last_jump):
            
            if curr_stone == last_stone:
                res.append(True)
                return
            
            if curr_stone > last_stone:
                res.append(False)
                return
            
            for i in range(-1,2):
                next_stone = curr_stone + (last_jump + i)
                if next_stone in stones and next_stone > curr_stone and (next_stone,last_jump+i) not in memo:
                    memo[(next_stone,last_jump+i)] = 1
                    dfs(next_stone,last_jump+i)
        
        dfs(1,stones[-1],1)
        return any(res)

有人可以帮助我解决这些问题吗?我总是在这些问题上苦苦挣扎,最终将值存储在数组中;但是,理想情况下,我希望递归代码的结果相同,而无需分配额外的 res 数组空间。

解决方法

由于该函数的整个目的似乎归结为返回 any(res),因此您似乎应该从递归函数中返回 True/False 而不是附加它们,然后退出一旦找到单个 True 值,则从所有递归调用中调用,而不必费心保存每个找到的值。

这将涉及检查从递归调用 dfs(next_stone,last_stone,last_jump+i) 返回的内容,如果为真,只需返回该 True

from typing import List

class Solution:
    def canCross(self,stones: List[int]) -> bool:

        if stones[1] != 1:
            return False

        memo = {}

        def dfs(curr_stone,last_jump):

            if curr_stone == last_stone:
                return True  # Return the results instead of appending them to a list

            if curr_stone > last_stone:
                return False

            for i in range(-1,2):
                next_stone = curr_stone + (last_jump + i)
                if next_stone in stones and next_stone > curr_stone and (next_stone,last_jump + i) not in memo:
                    memo[(next_stone,last_jump + i)] = 1
                    rec_result = dfs(next_stone,last_jump + i)
                    if rec_result:  # Then check the recursive results at the call-site
                        return True

        return dfs(1,stones[-1],1)

我会注意到,我没有对此进行广泛的测试,但从一些快速的“头脑解释”来看,它似乎是等效的。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。