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
廣告