1. 程式人生 > 實用技巧 >1262. 可被三整除的最大和

1262. 可被三整除的最大和

給你一個整數陣列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

來源:力扣(LeetCode)

連結:https://leetcode-cn.com/problems/greatest-sum-divisible-by-three

一拿到題,腦子裡浮出一些演算法,暴力,貪心,動規,,,

暴力=>回溯

萬物始於暴力

找出所有能被 3 整除的子集,選出和最大的。時間複雜度為 O(2^N) ,肯定超時。那就會想到回溯找子集

class Solution:
    def maxSumDivThree(self, nums: List[int]) -> int:
        self.res = 0
        def backtrack(temp, start):
            total 
= sum(temp) if total % 3 == 0: self.res = max(self.res, total) for i in range(start, len(nums)): temp.append(nums[i]) backtrack(temp, i + 1) temp.pop(-1) backtrack([], 0) return self.res

然而還是超時