糖逗 发表于 2021-4-3 18:43:31

C++刷leetcode(1262. 可被三整除的最大和***)【动态规划】

题目描述:
给你一个整数数组 nums,请你找出并返回能被三整除的元素最大和。

 

示例 1:

输入:nums =
输出:18
解释:选出数字 3, 6, 1 和 8,它们的和是 18(可被 3 整除的最大和)。
示例 2:

输入:nums =
输出:0
解释:4 不能被 3 整除,所以无法选出数字,返回 0。
示例 3:

输入:nums =
输出:12
解释:选出数字 1, 3, 4 以及 4,它们的和是 12(可被 3 整除的最大和)。
 

提示:

1 <= nums.length <= 4 * 10^4
1 <= nums <= 10^4

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/greatest-sum-divisible-by-three
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


class Solution {
public:
    int maxSumDivThree(vector<int>& nums) {
      //注意初始化!!!!!
      int len = nums.size();
      vector<vector<int> >dp(len, vector<int>(3, INT_MIN));
      //初始化
      if(nums % 3 == 0){
            dp = nums;
      }else if(nums % 3 == 1){
            dp = nums;
            dp = 0;
      }else{
            dp = nums;
            dp = 0;
      }
      //当num % 3 == 1时 dp并不等于0,但是dp可以等于0,因为0%3 = 0,而!=2;
      for(int i = 1; i < len; i++){
            int temp = nums;
            if(temp % 3 == 0){
                dp = max(dp, dp + temp);
                dp = max(dp, dp + temp);
                dp = max(dp, dp + temp);
            }else if(temp % 3 == 1){
                dp = max(dp, dp + temp);
                dp = max(dp, dp + temp);
                dp = max(dp, dp + temp);
            }else{
                dp = max(dp, dp + temp);
                dp = max(dp, dp + temp);
                dp = max(dp, dp + temp);
            }
      }
      return dp == INT_MIN ? 0 : dp;
    }
};

糖逗 发表于 2021-4-3 18:44:01

注意初始化!!!!{:10_262:}
页: [1]
查看完整版本: C++刷leetcode(1262. 可被三整除的最大和***)【动态规划】