给你两个数组 nums1 和 nums2 。
请你返回 nums1 和 nums2 中两个长度相同的 非空 子序列的最大点积。
数组的非空子序列是通过删除原数组中某些元素(可能一个也不删除)后剩余数字组成的序列,但不能改变数字间相对顺序。比方说,2,3,5 是 1,2,3,4,5 的一个子序列而 1,5,3 不是。
示例 1:
输入:nums1 = 2,1,-2,5, nums2 = 3,0,-6
输出:18
解释:从 nums1 中得到子序列 2,-2 ,从 nums2 中得到子序列 3,-6 。
它们的点积为 (23 + (-2)(-6)) = 18 。
示例 2:
输入:nums1 = 3,-2, nums2 = 2,-6,7
输出:21
解释:从 nums1 中得到子序列 3 ,从 nums2 中得到子序列 7 。
它们的点积为 (3*7) = 21 。
示例 3:
输入:nums1 = -1,-1, nums2 = 1,1
输出:-1
解释:从 nums1 中得到子序列 -1 ,从 nums2 中得到子序列 1 。
它们的点积为 -1 。
提示:
1 <= nums1.length, nums2.length <= 500
-1000 <= nums1i, nums2i <= 100
动态规划
cpp
class Solution {
public:
int maxDotProduct(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
// 初始化 dp 数组为一个极小的值 INT_MIN,因为点积可能为负数
vector<vector<int>> dp(m+1, vector<int>(n+1, INT_MIN));
for (int i = 1; i <= m; ++i) {
for (int j = 1; j <= n; ++j) {
int product = nums1[i-1] * nums2[j-1];
dp[i][j] = max({
product, // 只选择当前元素对
dp[i-1][j], // 不选择 nums1[i-1]
dp[i][j-1], // 不选择 nums2[j-1]
dp[i-1][j-1] == INT_MIN ? product : dp[i-1][j-1] + product // 检查溢出
});
}
}
return dp[m][n];
}
};
这道题是LCS的变形,dpij 表示使用 nums1 的前 i 个元素和 nums2 的前 j 个元素时,能够得到的最大点积。
在计算dpij的时候有四种情况:
一种是我们只选择product这个元素对,
一种是不选择当前元素对,由dpi-1j的状态转移而来。
一种是不选择当前元素对,由dpij-1的状态转移而来。
一种是选择当前元素对,并且对dpi-1j-1检查是否是INT_MIN情况,防止出现k为负数时的溢出。然后由之前的状态dpi-1j-1加上该元素对。
由于我们要求的是最大点积,那么就从这四种情况中选取最大的情况。
最后返回dpmn。