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