超级饮料的最大强化能量

题目

来自未来的体育科学家给你两个整数数组 energyDrinkA 和 energyDrinkB，数组长度都等于 n。这两个数组分别代表 A、B 两种不同能量饮料每小时所能提供的强化能量。

你需要每小时饮用一种能量饮料来 最大化 你的总强化能量。然而，如果从一种能量饮料切换到另一种，你需要等待一小时来梳理身体的能量体系（在那个小时里你将不会获得任何强化能量）。

返回在接下来的 n 小时内你能获得的 最大 总强化能量。

注意 你可以选择从饮用任意一种能量饮料开始。

示例 1：

输入：energyDrinkA = [1,3,1], energyDrinkB = [3,1,1]

输出：5

解释：

要想获得 5 点强化能量，需要选择只饮用能量饮料 A（或者只饮用 B）。

分析

这是个二维数组动态规划问题, d[i][0/1]表示i+1小时喝A/B能获得的最大能量

那么

d[i][0] = max([d[i-1][0], d[i-2][1]]) + energyDrinkA[i]
d[i][1] = max([d[i-1][1], d[i-2][0]]) + energyDrinkB[i]

结果返回max([d[n-1][0], d[n-1][1]])

解

class Solution:
    def maxEnergyBoost(self, energyDrinkA: List[int], energyDrinkB: List[int]) -> int:
        # 设d[i][0/1]表示第i天从A或者B取。
        # d[i][0] = max([d[i-1][0], d[i-2][1]]) + A[i]
        # d[i][1] = max([d[i-1][1], d[i-2][0]]) + B[i]
        n = len(energyDrinkA)
        d = [[0, 0] for _ in range(n)]
        d[0][0] = energyDrinkA[0]
        d[0][1] = energyDrinkB[0]

        for i in range(1, n):
            d[i][0] = max([d[i-1][0], d[i-2][1]]) + energyDrinkA[i]
            d[i][1] = max([d[i-1][1], d[i-2][0]]) + energyDrinkB[i]

        return max(d[n-1])