leetcode 3259.超级饮料的最大强化能量

是小Y啦2024-11-04 11:46

思路：dp

这里的dp方程其实很简单,题目中已经给出了，要么就是相邻的最大能量+当前的能量值，要么就是下一种饮料隔开一瓶饮料的最大能量值+当前能量值，取最大值即可。

注意：这里额外要注意的是，如果选择一行一行的进行dp递归，产生的结果不一定对，因为如果只是针对一种饮料的出发进行处理的话，另一种饮料的最大能量就没有赋值了；所以，我们需要对这两个饮料同时进行递归才有效。

不要忘记初始化res数组，让第一个饮料都赋值。

复制代码

class Solution {
    public long maxEnergyBoost(int[] energyDrinkA, int[] energyDrinkB) {
        long [][]res=new long[2][energyDrinkB.length];
        res[0][0]=energyDrinkA[0];
        res[1][0]=energyDrinkB[0];
        for(int i=1;i<energyDrinkA.length;i++){
            if(i-2>=0){
                res[0][i]=Math.max(res[1][i-2]+energyDrinkA[i],res[0][i-1]+energyDrinkA[i]);
                res[1][i]=Math.max(res[0][i-2]+energyDrinkB[i],res[1][i-1]+energyDrinkB[i]);
            }
            else{
                res[0][i]=res[0][i-1]+energyDrinkA[i];
                res[1][i]=res[1][i-1]+energyDrinkB[i];
            }
        }
        return (long)Math.max(res[0][energyDrinkA.length-1],res[1][energyDrinkB.length-1]);
    }
}