1262. 可被三整除的最大和

提示

中等

215

相关企业

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

Solution

设 tot\textit{tot}tot 是数组 nums\textit{nums}nums 中所有元素的和，此时 tot\textit{tot}tot 会有三种情况：

如果 tot\textit{tot}tot 是 333 的倍数，那么我们不需要丢弃任何数；

如果 tot\textit{tot}tot 模 333 余 111，此时我们有两种选择：要么丢弃 bbb 中最小的 111 个数，要么丢弃 ccc 中最小的 222 个数；

如果 tot\textit{tot}tot 模 333 余 222，此时我们有两种选择：要么丢弃 bbb 中最小的 222 个数，要么丢弃 ccc 中最小的 111 个数。

class Solution:
    def maxSumDivThree(self, nums: List[int]) -> int:
        s = sum(nums)
        d = defaultdict(list)
        tmp = 0
        for num in nums:
            d[num % 3].append(num)
        for i in range(3):
            d[i].sort()
        if s % 3 == 0:
            return s
        elif s % 3 == 1:
            if len(d[2]) > 1:
                tmp = s - d[2][0] - d[2][1]
            if d[1]:
                tmp = max(tmp, s - d[1][0])
        elif s % 3 == 2:
            if len(d[1]) > 1:
                tmp = s - d[1][0] - d[1][1]
            if d[2]:
                tmp = max(tmp, s - d[2][0])
        return tmp