跳转至

LeetCode: 1262. 可被三整除的最大和

1、题目描述

给你一个整数数组 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

2、解题思路

  • 动态规划
  • 初始化
初始化保存3个信息
pre[0,0,0]
分别对应 从数字开头到当前位置
- 最大的被3整除余0
- 最大的被3整除余1
- 最大的被3整除余2

然后每一个数字都去更新这三个状态

class Solution:
    def maxSumDivThree(self, nums: List[int]) -> int:
        pre = [0, 0, 0]

        for num in nums:
            remainder = num % 3
            a = pre[(3 + 0 - remainder) % 3]
            b = pre[(3 + 1 - remainder) % 3]
            c = pre[(3 + 2 - remainder) % 3]

            if a or remainder == 0:
                pre[0] = max(pre[0], a + num)
            if b or remainder == 1:
                pre[1] = max(pre[1], b + num)
            if c or remainder == 2:
                pre[2] = max(pre[2], c + num)
        return pre[0]