LeetCode 1340. Jump Game V

LeetCode 1340

这个问题，最直接的想法就是对每个位置都去求可能的jump次数，然后最后得到最大值。
对于每个位置的求解，我们可以向前向后走d的长度，找到单方向比本位置低的位置，然后自己的jump次数应该是这些值中最大的加1.
用memo可以来做一下优化，已经求解过的，我们不用重复求解。

    def maxJumps(self, arr: List[int], d: int) -> int:
        if len(arr) == 0: return 0
        dp = [1]* len(arr)
        memo = collections.defaultdict(int)

        def getJumpsAtPos(dp:List[int], arr:List[int], pos, d, memo):
            if pos in memo: return memo[pos]            
            for diff in range(1, d+1, 1):
                dpos = pos - diff
                if dpos >=0 and arr[pos] <= arr[dpos]:
                    break
                if dpos < 0: break
                dp[pos] = max(dp[pos],getJumpsAtPos(dp, arr, dpos, d, memo) + 1)

            for diff in range(1, d+1, 1):
                dpos = pos + diff
                if dpos < len(arr) and arr[pos] <= arr[dpos]:
                    break
                if dpos >= len(arr): break
                dp[pos] = max(dp[pos],getJumpsAtPos(dp, arr, dpos, d, memo) + 1)

            memo[pos] = dp[pos]
            return memo[pos]

        for i,v in enumerate(arr):
            getJumpsAtPos(dp, arr, i, d, memo)

        return max(dp)