题目描述:
给你一个整数数组 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的倍数那么就是最大,否则将数组按照从小到达排序,然后我们看求出的数组和sum mod 3等于多少(肯定不等于0,因为等于0
的话就直接返回了),比如我们可以从num数组中找到一个最小的数字,该数字满足mod 3 == mod,或者找到两个数字这两个数字mod 3都为 num mod 3的另一个(2 就是1 1就是2),如果找不到那么就是0
class Solution {
public int maxSumDivThree(int[] nums) {
Arrays.sort(nums);
if (null == nums || nums.length == 0) {
return 0;
}
if (nums.length == 1) {
if (nums[0] % 3 != 0) {
return 0;
} else
return nums[0];
}
long sum = 0;
for (int num : nums) {
sum += num;
}
if (sum % 3 == 0) {
return (int) sum;
}
return get(nums, (int) sum, (int) (sum % 3));
}
public int get(int[] nums, int sum, int mod) {
int min = 0;
int tems = 0;
boolean flag = true;
// 找一个mod看是否成立
for (int i = 0; i < nums.length; i++) {
int num = nums[i];
if (num % 3 == mod) {
flag = false;
tems = num;
break;
}
}
if (!flag) {
min = tems;
}
// 一个不成立,那么就找两个比如 2 那么找两个1来凑成2
mod = mod == 1 ? 2 : 1;
int tem = 0;
int two = 0;
for (int i = 0; i < nums.length; i++) {
int num = nums[i];
if (num % 3 == mod) {
two += num;
tem ++;
}
if(tem == 2){
break;
}
}
if(tem == 2 && !flag){
return sum - Math.min(min,two);
}else if(tem == 2){
return sum - two;
}else if(!flag){
return sum - min;
}
return 0;
}
}