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

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 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的长度大１，也就是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]即可

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