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)

你可能感兴趣的:(Leetcode)