leetcode 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 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

题目其实相当于字符串中的最长公共字串长度，使用二维数组记录每个位置对应的上一位是否匹配。如图所示 状态转移方程如下

if(a[i] == b[i]) dp[i][j] = dp[i-1][j-1]+1 else dp[i][j] = 0

代码

kotlin代码，为了防止i-1越界初始化二维数组长度+1

fun findLength(A: IntArray, B: IntArray): Int { var len = A.size var max = 0 // 初始化二维数组 var dp = Array(len + 1, { Array(len + 1) { it -> 0 } }) // 状态转移方程 if(A[j-1] == B[j-1]) dp[i][j] = dp[i-1][j-1]+1 else dp[i][j] = 0 for (i in 1..len) { for (j in 1..len) { if (A[i - 1] == B[j - 1]) dp[i][j] = dp[i - 1][j - 1] + 1 else dp[i][j] = 0 max = Math.max(dp[i][j], max) } } return max }