nums数组中的数可以分为三种类型s0,s1,s2;s0表示除以3余0的数,s1表示除以3余1的数;找出能被三整除的元素最大和sum,那么sum中包含全部s0,以及s1和s2组成的能被三整除的数(比如三个余1的数、一个余1一个余2的数、三个余2的数,等)。
如果sum中尽量包含更多的 s1和s2,那么最后的结果一共有5种情况:
- 剩余0个s1和s2(0)
- 剩余1个s1(1)
- 剩余1个s2(2)
- 剩余2个s1(1,1)
- 剩余2个s2(2,2)
设s1中有l1个数,s2中有l2个数。如果sum=s0+s1+s2(贪心算法),则sum%3= (l1+2*l2)%3=r
- r==0; 结果为情况1;return sum;
- r==1;结果只能为情况2或5;比较两种情况谁的元素和最大即可
- 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];
}
}
}
}
};