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 次觀看

開啟您的 職業生涯

完成課程以獲得認證

開始
廣告