leetcode中等1262可被三整除的最大和

Posted 2022-03-18 qq_40707462

给你一个整数数组 nums，请你找出并返回能被三整除的元素最大和。

示例 1：

输入：nums = [3,6,5,1,8]
输出：18
解释：选出数字 3, 6, 1 和 8，它们的和是 18（可被 3 整除的最大和）。

示例 2：

输入：nums = [4]
输出：0
解释：4 不能被 3 整除，所以无法选出数字，返回 0。

思路：类似三指针，a、b、c 分别记录此时余数为 0、1、2 时的元素最大和

class Solution 
    public int maxSumDivThree(int[] nums) 
        int a=0,b=0,c=0;
        int n=nums.length;
        for(int i=0;i<n;i++)
            if(nums[i]%3==0)
                a+=nums[i];
                b+=nums[i];
                c+=nums[i];
            
            if(nums[i]%3==1)
                int t=a;
                if((nums[i]+c)%3==0) a=Math.max(a,nums[i]+c);
                if((nums[i]+b)%3==2) c=Math.max(c,nums[i]+b);
                if((nums[i]+t)%3==1) b=Math.max(b,nums[i]+t);
            
            if(nums[i]%3==2)
                int t=a;
                if((nums[i]+b)%3==0) a=Math.max(a,nums[i]+b);
                if((nums[i]+c)%3==1) b=Math.max(b,nums[i]+c);
                if((nums[i]+t)%3==2) c=Math.max(c,nums[i]+t);
            
        
        return a;