leetcode的python实现-718. 最长重复子数组

    技术2022-07-10  137

    leetcode的python实现-718. 最长重复子数组

    题目描述 给两个整数数组 A 和 B ,返回两个数组中公共的、长度最长的子数组的长度。

    示例 1:

    输入: A: [1,2,3,2,1] B: [3,2,1,4,7] 输出: 3 解释: 长度最长的公共子数组是 [3, 2, 1]。

    说明: 1 <= len(A), len(B) <= 1000 0 <= A[i], B[i] < 100

    来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/maximum-length-of-repeated-subarray

    思路 本例属于动态规划类型的问题,设置dp二维数组,dp[i][j]表示A[i-1]和B[j-1]数字是否相同时的状态.当A[i-1] == B[j-1]时,dp[i][j] = dp[i-1][j-1] + 1,否则,dp[i][j]等于0.A为列B为行如下: A 3 2 1 4 7 B 1 0 0 1 0 0 2 0 1 0 0 0 3 1 0 0 0 0 2 0 2 0 0 0 1 0 0 3 0 0

    A和B都是从索引0开始搜索,但是dp[i][j] 与dp[i-1][j-1]相关,为保证i-1和j-1有意义,因此dp的维度比A和B的长度大1,也就是dp初始化为dp[len(A)+1][ len(B)+1],元素均为0.

    python3实现

    from typing import List​ class Solution: def findLength(self, A: List[int], B: List[int]) -> int: dp = [[0 for i in range( len(B) + 1)] for j in range( len(A) + 1)] maxlen = 0 # n = 0 for i in range(1,len(A)+1 ): for j in range(1,len(B)+1): if A[i-1] == B[j-1]: dp[i][j] = dp[i-1][j-1] + 1 if maxlen < dp[i][j]: maxlen = dp[i][j] # n = i # return A[n-maxlen : n] return maxlen

    如果要求返回最长的公共子数组序列,将上述程序中注释的代码取消注释,返回A[n-maxlen:n]即可

    欢迎关注公众号-算法学习总结,获取更多题解,欢迎共同交流学习

    Processed: 0.009, SQL: 9