Leetcode 3259. Maximum Energy Boost From Two Drinks

• Leetcode 3259. Maximum Energy Boost From Two Drinks
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3259. Maximum Energy Boost From Two Drinks

1. 解题思路

这一题就是一个动态规划，分别考察下两个数列的选择即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def maxEnergyBoost(self, energyDrinkA: List[int], energyDrinkB: List[int]) -> int:
        n = len(energyDrinkA)
        
        @lru_cache(None)
        def dp(idx, status):
            if idx >= n:
                return 0
            if status == "A":
                return max(energyDrinkA[idx] + dp(idx+1, "A"), dp(idx+1, "B"))
            elif status == "B":
                return max(dp(idx+1, "A"), energyDrinkB[idx] + dp(idx+1, "B"))
            else:
                return max(energyDrinkA[idx] + dp(idx+1, "A"), energyDrinkB[idx] + dp(idx+1, "B"))
            
        return dp(0, "")

提交代码评测得到：耗时2252ms，占用内存433.6MB。