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。