题目链接:最长重复子数组
题目描述:
给两个整数数组 A 和 B ,返回两个数组中公共的、长度最长的子数组的长度。
示例 1:
输入:
A: [1,2,3,2,1]
B: [3,2,1,4,7]
输出: 3
解释:
长度最长的公共子数组是 [3, 2, 1]。
题目分析:这道题和两个字符串的最长公共子序列类似,不同之处在于子数组是连续的,那么只需要改变一下递推公式即可,要求连续,因此在判断相同时累加,否则就直接赋值0(不同于最长公共子序列的取前面序列的最大值),这样才能保证连续性
假设两个数组分别为:
[1,2,3,4,2]
[2,3,2,1,4]
简单画下表格来对比一下吧
如果是最长子数组,max则为2(2,3)
dp[i][j] = A[i-1] == B[j-1] ? dp[i - 1][j - 1] + 1 : 0;
i/j | 1 | 2 | 3 | 4 | 2 |
---|---|---|---|---|---|
2 | 0 | 1 | 0 | 0 | 1 |
3 | 0 | 0 | 2 | 0 | 0 |
2 | 0 | 1 | 0 | 0 | 1 |
1 | 1 | 0 | 0 | 0 | 0 |
4 | 0 | 0 | 0 | 1 | 0 |
如果是最长子序列,max则为3(2,3,2/2,3,4)
if(A[i-1]==B[j-1])
dp[i][j] = dp[i-1][j-1]+1;
else
dp[i][j] = Math.max(dp[i-1][j],dp[i][j-1]);
i/j | 1 | 2 | 3 | 4 | 2 |
---|---|---|---|---|---|
2 | 0 | 1 | 1 | 1 | 1 |
3 | 0 | 1 | 2 | 2 | 3 |
2 | 0 | 1 | 2 | 2 | 3 |
1 | 1 | 1 | 2 | 2 | 3 |
4 | 0 | 1 | 2 | 3 | 3 |
对比可以看到,连续子数组只要求对角线上的值
代码:
class Solution {
public int findLength(int[] A, int[] B) {
int n = A.length, m = B.length;
int[][] dp = new int[n + 1][m + 1];
int ans = 0;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
dp[i][j] = A[i-1] == B[j-1] ? dp[i - 1][j - 1] + 1 : 0;
ans = Math.max(ans, dp[i][j]);
}
}
return ans;
}
}
这里把我写过的最长子序列贴一下方便对比:
最长子序列