C++ 中可被三整除的最大和


假设我们有一个整数数组 nums,我们需要找出给定数组中元素的最大可能和,该和必须是 3 的倍数。因此,如果输入类似于 [3,6,5,1,8],则输出将为 18,因为子数组是 [3,6,1,8],且和为 18,即 3 的倍数。

若要解决此问题,我们将遵循以下步骤:

  • n := nums 数组的大小
  • 创建大小为 (n + 1) x 3 的一个二维数组 dp
  • 设置 dp[0, 0] := 0,dp[0,1] := -inf,dp[0,2] := inf
  • 对 i 从 1 到 n 循环;
    • x := nums[i - 1]
    • 对 j 从 0 到 2 循环,dp[i, j] := dp[i – 1, j]
    • 对 j 从 0 到 2 循环
      • k := (x + j) mod 3
      • dp[i, k] := dp[i, k] 和 dp[i – 1, j] + x 的最大值
  • 返回 dp[n, 0]

让我们看看以下实现以获得更好的理解:

示例

 现场演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   int maxSumDivThree(vector<int>& nums) {
      int n = nums.size();
      int dp[n+1][3];
      dp[0][0] = 0;
      dp[0][1] = INT_MIN;
      dp[0][2] = INT_MIN;
      for(int i = 1; i <= n; i++){
         int x = nums[i-1];
         for(int j = 0; j < 3; j++)dp[i][j] = dp[i-1][j];
         for(int j = 0; j < 3; j++){
            int k = (x + j) % 3;
            dp[i][k] = max(dp[i][k],dp[i-1][j] + x);
         }
      }
      return dp[n][0];
   }
};
main(){
   vector<int> v = {3,6,5,1,8};
   Solution ob;
   cout << (ob.maxSumDivThree(v));
}

输入

[3,6,5,1,8]

输出

18

更新时间:02-5 月-2020

368 次观看

开启您的 职业生涯

完成课程以获得认证

开始
广告