给你一个整数数组 nums
,请你找出并返回能被三整除的元素最大和。
输入: nums = [3,6,5,1,8] 输出: 18 解释: 选出数字 3, 6, 1 和 8,它们的和是 18(可被 3 整除的最大和)。
输入: nums = [4] 输出: 0 解释: 4 不能被 3 整除,所以无法选出数字,返回 0。
输入: nums = [1,2,3,4,4] 输出: 12 解释: 选出数字 1, 3, 4 以及 4,它们的和是 12(可被 3 整除的最大和)。
1 <= nums.length <= 4 * 10^4
1 <= nums[i] <= 10^4
impl Solution {
pub fn max_sum_div_three(nums: Vec<i32>) -> i32 {
let mut min = vec![(19996, 10000), (20000, 10001)];
let mut ret = 0;
for num in nums {
ret += num;
match num as usize % 3 {
0 => (),
i => {
if num < min[i - 1].0 {
min[i - 1] = (num, min[i - 1].0);
} else if num < min[i - 1].1 {
min[i - 1].1 = num;
}
},
}
}
match ret % 3 {
1 => ret - min[0].0.min(min[1].0 + min[1].1),
2 => ret - min[1].0.min(min[0].0 + min[0].1),
_ => ret,
}
}
}