思路:
先把nums
排好序,
然后开一个字典,把nums
里的数按照对3取模的值进行分类,
然后会存在三种情况:
sum(nums) % 3 == 0
,此时无需任何处理,直接返回sum(nums)
即可。sum(nums) % 3 == 1
,此时有两种可能的解决办法: a. 删除一个模为1
的最小数;b. 删除模为2
的最小的两个数;选择删除之后剩余和最大的情况返回即可。sum(nums) % 3 == 2
,此时有两种可能的解决办法: a. 删除一个模为2
的最小数;b. 删除模为1
的最小的两个数;选择删除之后剩余和最大的情况返回即可。
代码实现:
class Solution(object):
def maxSumDivThree(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
from collections import defaultdict
dic = defaultdict(list)
nums.sort()
for num in nums:
dic[num % 3].append(num)
s = sum(nums)
if s % 3 == 0:
return s
if s % 3 == 2:
t1, t2 = float("inf"),float("inf")
if 2 in dic: #可以删除一个模为 2 的最小数
t1 = dic[2][0]
if len(dic[1]) >= 2: # 也可以删除模为 1 的最小的两个数
t2 = dic[1][0] + dic[1][1]
if t1 > t2:# 选择两种可能中较小的值删除
return s - t2
return s - t1
if s % 3 == 1:
t1, t2 = float("inf"), float("inf")
if 1 in dic: # 可以删除一个模为 1 的最小数
t1 = dic[1][0]
if len(dic[2]) >= 2: # 也可以删除模为 2 的最小的两个数
t2 = dic[2][0] + dic[2][1]
if t1 > t2: # 选择两种可能中较小的值删除
return s - t2
return s - t1
复杂度分析:
时间复杂度: O ( N l o g N ) O(NlogN) O(NlogN)
空间复杂度: O ( N ) O(N) O(N)
优化分析:
因为只需要找最小的两个值,所以其实可以不用排序,将时间复杂度降低到 O ( N ) O(N) O(N)。