琢磨ing,脑袋想不通
cpp
class Solution {
public:
int maxSumDivThree(vector<int>& nums) {
int n = nums.size(),
memo[n]
[3]; // memo[n][3]为二维数组,考虑数组中前 i+1 个元素(即从
// nums[0] 到 nums[i]),在所有可能的选择下,能得到的、除以
// 3 余数为 j 的最大子集和
// i 的范围是 0 到 n-1(代表数组中的每个元素)。
// j 的范围是 0, 1, 2(代表除以 3 可能得到的三种余数)。
// metset函数用来给一块内存区域赋初始值,这里把 memo
// 数组里的所有值都初始化为 -1,表示 memo[i][j] 这个状态我们还没有计算过
memset(memo, -1, sizeof(memo));
// 定义递归函数(Lambda 表达式)
//[&] 是捕获列表。& 表示按引用捕获外部作用域的所有变量(比如 nums, n,
//memo)
auto dfs = [&](auto&& dfs, int i, int j) -> int {
// auto&& dfs让这个 Lambda 可以递归调用自己。dfs 既是这个 Lambda
// 表达式本身的名字,也作为参数传递进来。
//-> int:指定了这个 Lambda 表达式的返回值是 int 类型。
if (i < 0)
return j ? INT_MIN : 0;
// 定义一个引用 res,让它指向 memo[i][j]。这样,对 res
// 的操作就等同于对 memo[i][j] 的操作。
int& res = memo[i][j];
if (res != -1)
// 如果 res 不等于 -1,说明我们之前已经计算过 "考虑前 i+1
// 个元素,余数为 j"
// 的最大和了。我们直接返回这个存储好的结果,不再进行重复的递归计算
return res;
return res = max(dfs(dfs, i - 1, j),
dfs(dfs, i - 1, (j + nums[i]) % 3) + nums[i]);
};
return dfs(dfs, n - 1, 0); // 这是整个函数的入口调用
// i = n - 1:表示我们要考虑数组中的最后一个元素(即所有元素)。
// j = 0:表示我们的目标是找到一个子集,其和能被 3 整除(余数为 0)。
}
};