可被三整除的最大和【力扣1262】

nums数组中的数可以分为三种类型s0,s1,s2;s0表示除以3余0的数,s1表示除以3余1的数;找出能被三整除的元素最大和sum,那么sum中包含全部s0,以及s1和s2组成的能被三整除的数(比如三个余1的数、一个余1一个余2的数、三个余2的数,等)。

如果sum中尽量包含更多的 s1和s2,那么最后的结果一共有5种情况:

  1. 剩余0个s1和s2(0)
  2. 剩余1个s1(1)
  3. 剩余1个s2(2)
  4. 剩余2个s1(1,1)
  5. 剩余2个s2(2,2)

设s1中有l1个数,s2中有l2个数。如果sum=s0+s1+s2(贪心算法),则sum%3= (l1+2*l2)%3=r

  1. r==0; 结果为情况1;return sum;
  2. r==1;结果只能为情况2或5;比较两种情况谁的元素和最大即可
  3. r==2;结果只能为情况3或4;比较两种情况谁的元素和最大即可

代码示例:

class Solution {
public:
    static bool cmp(int x,int y){
        return x<y;
    }
    int maxSumDivThree(vector<int>& nums) {
        vector<int> s1;
        vector<int> s2;
        long long int sum=0;
        //对数组进行分类
        for(int i=0;i<nums.size();i++){
            sum+=nums[i];
            if(nums[i]%3==1){
                s1.push_back(nums[i]);
            }else if(nums[i]%3==2){
                s2.push_back(nums[i]);
            }
        }
        sort(s1.begin(),s1.end(),cmp);
        sort(s2.begin(),s2.end(),cmp);
        int l1,l2;
        l1=s1.size();
        l2=s2.size();
        if((l1+2*l2)%3==0){                       //r==0
            return sum;
        }else if((l1+2*l2)%3==1){                 //r==1
            if(l1==0&&l2<2){                            //情况2、5都不成立
                return 0;
            }else if(l1==0&&l2>=2){                     //情况5成立
                return sum-s2[0]-s2[1];
            }else if(l1>0&&l2<2){                       //情况2成立
                return sum-s1[0];
            }else{                                      //情况2、5都成立,比较谁的结果更大
                if(s1[0]<s2[0]+s2[1]){
                    return sum-s1[0];
                }else{
                    return sum-s2[0]-s2[1];
                }
            }
        }else{                                   //r==2
            if(l1<2&&l2<1){
                return 0;
            }else if(l1>=2&&l2<1){
                return sum-s1[0]-s1[1];
            }else if(l1<2&&l2>=1){
                return sum-s2[0];
            }else{
                if(s1[0]+s1[1]<s2[0]){
                    return sum-s1[0]-s1[1];
                }else{
                    return sum-s2[0];
                }
            }
        }
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值