C++刷leetcode（1262. 可被三整除的最大和***）【动态规划】

糖逗

题目描述：

给你一个整数数组 nums，请你找出并返回能被三整除的元素最大和。

 

示例 1：

输入：nums = [3,6,5,1,8]
输出：18
解释：选出数字 3, 6, 1 和 8，它们的和是 18（可被 3 整除的最大和）。
示例 2：

输入：nums = [4]
输出：0
解释：4 不能被 3 整除，所以无法选出数字，返回 0。
示例 3：

输入：nums = [1,2,3,4,4]
输出：12
解释：选出数字 1, 3, 4 以及 4，它们的和是 12（可被 3 整除的最大和）。
 

提示：

1 <= nums.length <= 4 * 10^4
1 <= nums[i] <= 10^4

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/greatest-sum-divisible-by-three
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
    int maxSumDivThree(vector<int>& nums) {
        //注意初始化！！！！！
        int len = nums.size();
        vector<vector<int> >dp(len, vector<int>(3, INT_MIN));
        //初始化
        if(nums[0] % 3 == 0){
            dp[0][0] = nums[0];
        }else if(nums[0] % 3 == 1){
            dp[0][1] = nums[0];
            dp[0][0] = 0;
        }else{
            dp[0][2] = nums[0];
            dp[0][0] = 0;
        }
        //当num[0] % 3 == 1时 dp[0][2]并不等于0，但是dp[0][0]可以等于0，因为0%3 = 0，而！=2；
        for(int i = 1; i < len; i++){
            int temp = nums[i];
            if(temp % 3 == 0){
                dp[i][0] = max(dp[i-1][0], dp[i-1][0] + temp);
                dp[i][1] = max(dp[i-1][1], dp[i-1][1] + temp);
                dp[i][2] = max(dp[i-1][2], dp[i-1][2] + temp);
            }else if(temp % 3 == 1){
                dp[i][0] = max(dp[i-1][0], dp[i-1][2] + temp);
                dp[i][1] = max(dp[i-1][1], dp[i-1][0] + temp);
                dp[i][2] = max(dp[i-1][2], dp[i-1][1] + temp);
            }else{
                dp[i][0] = max(dp[i-1][0], dp[i-1][1] + temp);
                dp[i][1] = max(dp[i-1][1], dp[i-1][2] + temp);
                dp[i][2] = max(dp[i-1][2], dp[i-1][0] + temp);
            }
        }
        return dp[len-1][0] == INT_MIN ? 0 : dp[len-1][0];
    }
};

糖逗

注意初始化！！！！