题目
给两个整数数组 A 和 B ,返回两个数组中公共的、长度最长的子数组的长度。
示例:
输入:
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[i][j]表示A数组以i结尾,B数组以j结尾时,AB的最长公共子数组长度为dp[i][j]。
if (A[i] == B[j])
dp[i][j] = dp[i-1][j-1] + 1
else
dp[i][j] = 0
这里有两个问题
- 为什么dp[i][j]和dp[i-1][j]、dp[i][j-1]没关系
- 为什么要以下标结尾
第一个问题是,在遍历过程中是从左到右,从上到下,已经遍历了所有情况了。
第二个问题是,以结尾能明确的表示状态转移方程,如果不定义结尾,状态转移方程无法定义。
Base
由状态转移方程可知
int[][] dp = new int[A.length][B.length];
dp[0][0] = A[0] == B[0] ? 1 : 0;
for (int i = 0; i < A.length; i++) {
dp[i][0] = A[i] == B[0] ? 1 : 0;
}
for (int i = 0; i < B.length; i++) {
dp[0][i] = A[0] == B[i] ? 1 : 0;
}
答案
class Solution {
public int findLength(int[] A, int[] B) {
int[][] dp = new int[A.length][B.length];
dp[0][0] = A[0] == B[0] ? 1 : 0;
for (int i = 0; i < A.length; i++) {
dp[i][0] = A[i] == B[0] ? 1 : 0;
}
for (int i = 0; i < B.length; i++) {
dp[0][i] = A[0] == B[i] ? 1 : 0;
}
int result = 0;
for (int i = 1; i < A.length; i++) {
for (int j = 1; j < B.length; j++) {
if (A[i] == B[j])
dp[i][j] = dp[i - 1][j - 1] + 1;
else
dp[i][j] = 0;
result = Math.max(result, dp[i][j]);
}
}
return result;
}
}