题目描述:给你两个数组 nums1 和 nums2 。
请你返回 nums1 和 nums2 中两个长度相同的 非空 子序列的最大点积。
数组的非空子序列是通过删除原数组中某些元素(可能一个也不删除)后剩余数字组成的序列,但不能改变数字间相对顺序。比方说,[2,3,5] 是 [1,2,3,4,5] 的一个子序列而 [1,5,3] 不是。
解题思路:动态规划dp[i][j]
表示nums1[:i+1]
和nums2[:j+1]
的长度相同的非空子序列的最大点积,他的结果一共分为三种情况(1)最大点积没有用到nums1[i]
,那么结果就是dp[i-1][j]
(2)如果最大点积没有用到nums2[j]
,那么结果就是dp[i][j-1]
,(3)同时用到了nums1[i]和nums2[j]
,那么结果就是nums1[i]*nums2[j]再加上dp[i-1][j-1],如果它大于零的话
,代码如下:
class Solution:
def maxDotProduct(self, nums1: List[int], nums2: List[int]) -> int:
size1 = len(nums1)
size2 = len(nums2)
dp = [[0] * size2 for _ in range(size1)]
dp[0][0] = nums1[0] * nums2[0]
for i in range(1, size2):
dp[0][i] = max(dp[0][i-1], nums2[i]*nums1[0])
for i in range(1, size1):
dp[i][0] = max(dp[i-1][0], nums1[i]*nums2[0])
for i in range(1, size1):
for j in range(1, size2):
dp[i][j] = max(dp[i-1][j], dp[i][j-1], nums1[i]*nums2[j]+(dp[i-1][j-1] if dp[i-1][j-1] > 0 else 0))
return dp[-1][-1]