leetcode 1262. 可被三整除的最大和（C++、python）

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

思路：

-先求出数组的和，然后统计被3除余1和余2的数；

-若数组的和被3除余1，则减去1个被3除余1的数，或减去2个被3除余2的数，取较大者

-若数组的和被3除余2，则减去1个被3除余2的数，或减去2个被3除余1的数，取较大者

C++

class Solution {
public:
    int maxSumDivThree(vector<int>& nums) 
    {
        int n=nums.size();
        int sum=accumulate(nums.begin(),nums.end(),0);
        vector<int> tmp1;
        vector<int> tmp2;
        for(auto it:nums)
        {
            if(1==it%3)
            {
                tmp1.push_back(it);
            }
            else if(2==it%3)
            {
                tmp2.push_back(it);
            }
        }
        sort(tmp1.begin(),tmp1.end());
        sort(tmp2.begin(),tmp2.end());
        if(0==sum%3)
        {
            return sum;
        }
        else if(1==sum%3)
        {
            int v1=sum;
            int v2=sum;
            if(tmp1.size()>0)
            {
                v1-=tmp1[0];
            }
            if(tmp2.size()>1)
            {
                v2-=(tmp2[0]+tmp2[1]);
            }
            if(v1<sum && v2<sum)
            {
               sum=max(v1,v2); 
            }
            else
            {
                if(v1<sum)
                {
                    sum=v1;
                }
                else
                {
                    sum=v2;
                }
            }
        }
        else if(2==sum%3)
        {
            int v1=sum;
            int v2=sum;
            if(tmp1.size()>1)
            {
                v1-=(tmp1[0]+tmp1[1]);
            }
            if(tmp2.size()>0)
            {
                v2-=tmp2[0];
            }
            if(v1<sum && v2<sum)
            {
               sum=max(v1,v2); 
            }
            else
            {
                if(v1<sum)
                {
                    sum=v1;
                }
                else
                {
                    sum=v2;
                }
            }
        }
        return sum;
    }
};

python

class Solution:    
    def help(self,v1,v2,val):
        if v1<val and v2<val:
            return max(v1,v2)
        else:
            if v1<val:
                return v1
            else:
                return v2
    
    def maxSumDivThree(self, nums: List[int]) -> int:
        n=len(nums)
        val=sum(nums)
        tmp1=[]
        tmp2=[]
        for c in nums:
            if 1==c%3:
                tmp1.append(c)
            elif 2==c%3:
                tmp2.append(c)
        tmp1.sort()
        tmp2.sort()
        if 1==val%3:
            v1=val
            v2=val
            if len(tmp1)>0:
                v1-=tmp1[0]
            if len(tmp2)>1:
                v2-=(tmp2[0]+tmp2[1])
            val=self.help(v1,v2,val)
        elif 2==val%3:
            v1=val
            v2=val
            if len(tmp1)>1:
                v1-=(tmp1[0]+tmp1[1])
            if len(tmp2)>0:
                v2-=tmp2[0]
            val=self.help(v1,v2,val)
        return val