1458. 两个子序列的最大点积
题目
给你两个数组 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 。
思路
直接定义状态和转移方程
定义f[i][j]表示num1中前i个数字和num2中前j个数字所能得到的最大点积,那么转移方程有以下几种情况
- 不选择num1的第i个元素,那么就是f[i-1][j]
- 不选择num2的第j个元素,那么就是f[i][j-1]
- 选择num1的第i个元素和num2的第j个元素,那么就是f[i-1][j-1]+num1[i]*num2[j]
- 只选择num1的第i个元素和num2的第j个元素,就是num1[i]*num2[j](这种情况一开始被我忽略了,确实不容易想到)
或者可以换一个角度理解,见
https://leetcode-cn.com/problems/max-dot-product-of-two-subsequences/solution/cong-lcstui-dao-de-jie-fa-by-gk_climber/
代码
public int maxDotProduct(int[] nums1, int[] nums2) {
int n1 = nums1.length;
int n2 = nums2.length;
int f[][] = new int[n1+1][n2+1];
for(int i=1;i<=n1;i++){
for(int j=1;j<=n2;j++){
if(i==1 && j==1){
f[i][j] = nums1[i-1]*nums2[j-1];
}else if(i==1&&j!=1){
f[i][j] = Math.max(f[i][j-1],nums1[i-1]*nums2[j-1]);
}else if(i!=1&&j==1){
f[i][j] = Math.max(f[i-1][j],nums1[i-1]*nums2[j-1]);
}else{
f[i][j] = Math.max(f[i-1][j],f[i][j-1]);
f[i][j] = Math.max(f[i-1][j-1]+nums1[i-1]*nums2[j-1],f[i][j]);
f[i][j] = Math.max(nums1[i-1] * nums2[j-1], f[i][j]);
}
}
}
return f[n1][n2];
}