LeetCode 1262.可被三整除的最大和
程序员文章站
2022-03-04 23:52:22
...
给你一个整数数组 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
笔记思路:
动态规划的思路:主要是动态方程的寻找。
首先定义dp数组,i 表示的是当前数组中的到哪一个数字了。。。j 表示的是当前这个i的%3的余数是几
比如说:当前该数模三是1,那么dp[i][0]希望该数加上之前dp【i-1】【2】模三是2的和最好
dp数组存放的是当前i的最大值及其对应的模三的余数。
当前数字模三是多少,我希望能加上之前模三为三的差的数字和最好
class Solution {
public int maxSumDivThree(int[] nums) {
int[] dp = new int[3]; //数组初始化
dp[0] = 0;
dp[1] = Integer.MIN_VALUE;
dp[2] = Integer.MIN_VALUE;
for (int i = 0; i < nums.length; i++) {
int k = nums[i] % 3;
int[] dp_temp = dp.clone();
for (int j = 0; j < 3; j++) {
dp[j] = Math.max(dp_temp[j], dp_temp[(j-k+3)%3] + nums[i]);
}
}
return dp[0];
}
}