LeetCode 1262 可被3整除的最大和 解题思路

题目：

给你一个整数数组 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 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

根据题意很容易想到用状态转移与动态规划的思路来解决

定义

dp[i][0]表示nums[0...i]模三余零的最大和

dp[i][1]表示nums[0...i]模三余一的最大和

dp[i][2]表示nums[0...i]模三余二的最大和

零状态：当前数字最大和模三余零

一状态：当前数字最大和模三余一

二状态：当前数字最大和模三余二

动态规划的思路

对于任意一种状态，下一步我们都有两种选择，一是选择前当元素，二是不选择当前元素

dp[i][*] = max{dp[i-1][*],dp[i-1][*] + nums[i]}  (* 取值为 0,1,2)

以上是常见的动态规划的递推结构

状态转移

本题的状态转移显而易见，以当前状态是零状态为例。我们可以想到，前一个状态无非是零状态、一状态、二状态，三种情况，针对这三种情况我们分类讨论即可

动态规划与状态转移结合 显然可以直接两种方法直接结合起来

所以零状态如何转移我们理解了之后，可以一次写出一状态的转移，二状态的转移

示例：

为了便于理解此状态转移方程，进行示例：

如果二维数组有第 i 行是 [6, 10, 8] ，此时分别代表三个状态 0 1 2

写下一行（i+1）时，同样要保证下一行（i+1）为0， 1， 2的状态

此时的正整数为4 4mod3==1 属于1状态

那么填写下一行（i+1）的第一个数（代表0状态）时， 你要保证这个正整数加上这一行（i）的某一个数，为0状态。

那么1状态的数加2状态为0状态，所以4应该加8==12作为0状态

写i+1行的第二个数时，这个数应该是1状态，4为1状态，他应该加0状态，所以4应该加6==10 为1状态

写i+1行的第三个数时，这个数应该为2状态，4为1状态，他应该加1状态，所以4应该加10==14 为2状态

i+1 行的结果为 [12 ,10 ,14]

代码(python)：

class Solution(object):
    def maxSumDivThree(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
​
        dp = [[0]*3 for _ in range(len(nums)+1)]
        dp[0][1] = -float('inf')
        dp[0][2] = -float('inf')
        for i in range(len(nums)):
            if nums[i]%3==0:
                dp[i+1][0] = max(dp[i][0], dp[i][0]+nums[i])
                dp[i+1][1] = max(dp[i][1], dp[i][1]+nums[i])
                dp[i+1][2] = max(dp[i][2], dp[i][2]+nums[i])
            elif nums[i]%3==1:
                dp[i+1][0] = max(dp[i][0], dp[i][2]+nums[i])
                dp[i+1][1] = max(dp[i][1], dp[i][0]+nums[i])
                dp[i+1][2] = max(dp[i][2], dp[i][1]+nums[i])
            else:
                dp[i+1][0] = max(dp[i][0], dp[i][1]+nums[i])
                dp[i+1][1] = max(dp[i][1], dp[i][2]+nums[i])
                dp[i+1][2] = max(dp[i][2], dp[i][0]+nums[i])
        return dp[-1][0]

状态转移方程式及部分题解来源于：

作者：christmas_wang 链接：https://leetcode-cn.com/problems/greatest-sum-divisible-by-three/solution/dong-tai-gui-hua-yu-zhuang-tai-zhuan-yi-by-christm/ 来源：力扣（LeetCode）