LeetCode-Python-1262. 可被三整除的最大和(哈希表 + 数学)

思路:

先把nums排好序,

然后开一个字典,把nums里的数按照对3取模的值进行分类,

然后会存在三种情况:

  1. sum(nums) % 3 == 0,此时无需任何处理,直接返回sum(nums)即可。
  2. sum(nums) % 3 == 1,此时有两种可能的解决办法: a. 删除一个模为1的最小数;b. 删除模为2的最小的两个数;选择删除之后剩余和最大的情况返回即可。
  3. 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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值