Leetcode 1262. Greatest Sum Divisible by Three（一题容易错的DP题）

Leetcode 1262. Greatest Sum Divisible by Three

题目链接： Greatest Sum Divisible by Three

难度：Medium

题目大意：

选取数组中的若干个元素，使得这些元素的和能被3整除且尽可能得大。

思路：

01背包的变形，注意细节。

代码

class Solution {
    public int maxSumDivThree(int[] nums) {
        int n=nums.length;
        int M=3;
        int[] dp=new int[M];
        for(int i=1;i<=n;i++){
            int[] temp=Arrays.copyOf(dp,M);
            for(int j=0;j<M;j++){
                int mod=(temp[j]+nums[i-1])%M;//这里要用temp[j]的值，dp[j]可能会更新
                dp[mod]=Math.max(dp[mod],temp[j]+nums[i-1]);//这里要用temp[j]的值，dp[j]可能会更新
                //一维dp，如果mod没有取到就保持不变，比如对某个nums[i-1],j=[1,M)，
                //mod都是0，那么dp[1],dp[2]就保持不变就好了
                //如果用二维dp，还需要让dp[1],dp[2]等于dp[i-1][1]和dp[i-1][2]而不是初始值，反而比较麻烦,这个bug找了一个多小时才找到
            }
        }
        return dp[0];
    }
}