Leetcode.1262 可被三整除的最大和

题目链接

Leetcode.1262 可被三整除的最大和 Rating ： 1762

题目描述

给你一个整数数组 $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\ast 10^4$
• $1<=nums[i]<=10^4$

解法：线性dp

定义 $f(i,j)$ 为 处理前 $i$ 个元素，余数为 $j$ 的最大和。按照定义，最终的答案就为 $f(n,0)$；

对于 $nums[i-1]$，可选可不选。

如果 $nums[i-1]\%3=0$，那么说明加入当前 $nums[i-1]$ 不会改变其余数，所以：

• f [ i ] [ 0 ] = m a x { f [ i − 1 ] [ 0 ] , f [ i − 1 ] [ 0 ] + n u m s [ i − 1 ] } f[i][0] = max \{ f[i - 1][0] , f[i - 1][0] + nums[i - 1] \} f[i][0]=max{f[i−1][0],f[i−1][0]+nums[i−1]}；
• f [ i ] [ 1 ] = m a x { f [ i − 1 ] [ 1 ] , f [ i − 1 ] [ 1 ] + n u m s [ i − 1 ] } f[i][1] = max \{ f[i - 1][1] , f[i - 1][1] + nums[i - 1] \} f[i][1]=max{f[i−1][1],f[i−1][1]+nums[i−1]}；
• f [ i ] [ 2 ] = m a x { f [ i − 1 ] [ 2 ] , f [ i − 1 ] [ 2 ] + n u m s [ i − 1 ] } f[i][2] = max \{ f[i - 1][2] , f[i - 1][2] + nums[i - 1] \} f[i][2]=max{f[i−1][2],f[i−1][2]+nums[i−1]}；

如果 $nums[i-1]\%3=1$，那么说明加入当前 $nums[i-1]$ 之后，原来的余数 $0$ 会变成 $1$，$1$ 变成 $2$，$2$ 变成 $0$：

• f [ i ] [ 0 ] = m a x { f [ i − 1 ] [ 0 ] , f [ i − 1 ] [ 2 ] + n u m s [ i − 1 ] } f[i][0] = max \{ f[i - 1][0] , f[i - 1][2] + nums[i - 1] \} f[i][0]=max{f[i−1][0],f[i−1][2]+nums[i−1]}；
• f [ i ] [ 1 ] = m a x { f [ i − 1 ] [ 1 ] , f [ i − 1 ] [ 0 ] + n u m s [ i − 1 ] } f[i][1] = max \{ f[i - 1][1] , f[i - 1][0] + nums[i - 1] \} f[i][1]=max{f[i−1][1],f[i−1][0]+nums[i−1]}；
• f [ i ] [ 2 ] = m a x { f [ i − 1 ] [ 2 ] , f [ i − 1 ] [ 1 ] + n u m s [ i − 1 ] } f[i][2] = max \{ f[i - 1][2] , f[i - 1][1] + nums[i - 1] \} f[i][2]=max{f[i−1][2],f[i−1][1]+nums[i−1]}；

如果 $nums[i-1]\%3=2$，那么说明加入当前 $nums[i-1]$ 之后，原来的余数 $0$ 会变成 $2$，$1$ 变成 $0$，$2$ 变成 $1$：

• f [ i ] [ 0 ] = m a x { f [ i − 1 ] [ 0 ] , f [ i − 1 ] [ 1 ] + n u m s [ i − 1 ] } f[i][0] = max \{ f[i - 1][0] , f[i - 1][1] + nums[i - 1] \} f[i][0]=max{f[i−1][0],f[i−1][1]+nums[i−1]}；
• f [ i ] [ 1 ] = m a x { f [ i − 1 ] [ 1 ] , f [ i − 1 ] [ 2 ] + n u m s [ i − 1 ] } f[i][1] = max \{ f[i - 1][1] , f[i - 1][2] + nums[i - 1] \} f[i][1]=max{f[i−1][1],f[i−1][2]+nums[i−1]}；
• f [ i ] [ 2 ] = m a x { f [ i − 1 ] [ 2 ] , f [ i − 1 ] [ 0 ] + n u m s [ i − 1 ] } f[i][2] = max \{ f[i - 1][2] , f[i - 1][0] + nums[i - 1] \} f[i][2]=max{f[i−1][2],f[i−1][0]+nums[i−1]}；

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    int maxSumDivThree(vector<int>& nums) {
        int n = nums.size();
        int f[n + 1][3];

        memset(f,0,sizeof f);
        f[0][1] = -1e9 , f[0][2] = -1e9;

        for(int i = 1;i <= n;i++){
            int r = nums[i - 1] % 3;
            if(r == 0){
                f[i][0] = max(f[i - 1][0] , f[i - 1][0] + nums[i - 1]);
                f[i][1] = max(f[i - 1][1] , f[i - 1][1] + nums[i - 1]);
                f[i][2] = max(f[i - 1][2] , f[i - 1][2] + nums[i - 1]);
            }
            else if(r == 1){
                f[i][0] = max(f[i - 1][0] , f[i - 1][2] + nums[i - 1]);
                f[i][1] = max(f[i - 1][1] , f[i - 1][0] + nums[i - 1]);
                f[i][2] = max(f[i - 1][2] , f[i - 1][1] + nums[i - 1]);
            }
            else{
                f[i][0] = max(f[i - 1][0] , f[i - 1][1] + nums[i - 1]);
                f[i][1] = max(f[i - 1][1] , f[i - 1][2] + nums[i - 1]);
                f[i][2] = max(f[i - 1][2] , f[i - 1][0] + nums[i - 1]);
            }
        }
        return f[n][0];
    }
};

简化版：

class Solution {
public:
    int maxSumDivThree(vector<int>& nums) {
        int n = nums.size();
        int f[n + 1][3];

        memset(f,0,sizeof f);
        f[0][1] = -1e9 , f[0][2] = -1e9;

        for(int i = 1;i <= n;i++){
            int r = nums[i - 1] % 3;

            f[i][0] = max(f[i - 1][0] , f[i-1][(3 - r) % 3] + nums[i - 1]);
            f[i][1] = max(f[i - 1][1] , f[i-1][(3 - r + 1) % 3] + nums[i - 1]);
            f[i][2] = max(f[i - 1][2] , f[i-1][(3 - r + 2) % 3] + nums[i - 1]);
        }
        return f[n][0];
    }
};