leetcode 1458. 两个子序列的最大点积 困难

给你两个数组 nums1nums2

请你返回 nums1nums2 中两个长度相同的 非空 子序列的最大点积。

数组的非空子序列是通过删除原数组中某些元素(可能一个也不删除)后剩余数字组成的序列,但不能改变数字间相对顺序。比方说,[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] 。
它们的点积为 (2*3 + (-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 <= nums1[i], nums2[i] <= 1000

点积:

复制代码
定义 a = [a1, a2,..., an] 和b = [b1, b2,..., bn] 的点积为:
复制代码
这里的 Σ 指示总和符号。

分析:动态规划,令 dp[i][j] 代表 nums1 前 i 个数,nums2 前 j 个数这两个子序列可以取得的最大点积,可初始化 dp 数组的第一行和第一列为:

dp[1][1]=nums1[0]*nums2[0],

dp[1][i]=max(dp[1][i-1],nums1[0]*nums2[i-1]),

dp[i][1]=max(dp[i-1][1],nums1[i-1]*nums2[0])。

这里代表其中一个数组只取第一个数,另一个数组逐渐取到所有数时可以获得的最大子序列点积。当 i 和 j 都大于 1 时, dp[i][j] 可以由 dp[i-1][j],dp[i][j-1],以及 dp[i-1][j-1] 转移而来,具体来说,即 nums1 前 i 个数,nums2 前 j 个数这两个子序列可以取得的最大点积,要么是:

nums1 前 i-1 个数,nums2 前 j 个数的最大点积,或者

nums1 前 i 个数,nums2 前 j-1 个数的最大点积,或者

nums1 前 i-1 个数,nums2 前 j-1 个数的最大点积加上 nums[i]*nums[j]。

可得转移方程:dp[i][j]=max(dp[i-1][j-1]+nums1[i-1]*nums2[j-1],max(dp[i-1][j],dp[i][j-1]))

注意如果 dp[i-1][j-1] 小于零,而 nums1[i-1]*nums2[j-1] 大于 0,那么转移方程中 dp[i-1][j-1]+nums1[i-1]*nums2[j-1] 就要变成 nums1[i-1]*nums2[j-1]。

cpp 复制代码
int maxDotProduct(int* nums1, int nums1Size, int* nums2, int nums2Size) {
    int ans=0;
    int dp[510][510];
    for(int i=0;i<=500;++i)
        for(int j=0;j<=500;++j)
            dp[i][j]=0;
    dp[1][1]=nums1[0]*nums2[0];
    for(int i=2;i<=nums2Size;++i)
        dp[1][i]=fmax(dp[1][i-1],nums1[0]*nums2[i-1]);
    for(int i=2;i<=nums1Size;++i)
        dp[i][1]=fmax(dp[i-1][1],nums1[i-1]*nums2[0]);
    
    for(int i=2;i<=nums1Size;++i)
    {
        for(int j=2;j<=nums2Size;++j)
        {
            int sum=nums1[i-1]*nums2[j-1];
            if(sum>0&&dp[i-1][j-1]<0)dp[i][j]=fmax(sum,fmax(dp[i-1][j],dp[i][j-1]));
            else dp[i][j]=fmax(dp[i-1][j-1]+sum,fmax(dp[i-1][j],dp[i][j-1]));
        }
    }

    return dp[nums1Size][nums2Size];
}
相关推荐
醉颜凉41 分钟前
【LeetCode】打家劫舍III
c语言·算法·leetcode·树 深度优先搜索·动态规划 二叉树
达文汐44 分钟前
【困难】力扣算法题解析LeetCode332:重新安排行程
java·数据结构·经验分享·算法·leetcode·力扣
一匹电信狗44 分钟前
【LeetCode_21】合并两个有序链表
c语言·开发语言·数据结构·c++·算法·leetcode·stl
User_芊芊君子44 分钟前
【LeetCode经典题解】搞定二叉树最近公共祖先:递归法+栈存路径法,附代码实现
算法·leetcode·职场和发展
培风图南以星河揽胜1 小时前
Java版LeetCode热题100之零钱兑换:动态规划经典问题深度解析
java·leetcode·动态规划
算法_小学生1 小时前
LeetCode 热题 100(分享最简单易懂的Python代码!)
python·算法·leetcode
执着2591 小时前
力扣hot100 - 234、回文链表
算法·leetcode·链表
熬夜造bug1 小时前
LeetCode Hot100 刷题路线(Python版)
算法·leetcode·职场和发展
老鼠只爱大米1 小时前
LeetCode经典算法面试题 #108:将有序数组转换为二叉搜索树(递归分治、迭代法等多种实现方案详解)
算法·leetcode·二叉树·二叉搜索树·平衡树·分治法
踩坑记录1 小时前
leetcode hot100 104. 二叉树的最大深度 easy 递归dfs 层序遍历bfs
leetcode·深度优先·宽度优先