题目描述:
Given two integer arrays A
and B
, return the maximum length of an subarray that appears in both arrays.
Example 1:
Input:
A: [1,2,3,2,1]
B: [3,2,1,4,7]
Output: 3
Explanation:
The repeated subarray with maximum length is [3, 2, 1].
Note:
- 1 <= len(A), len(B) <= 1000
- 0 <= A[i], B[i] < 100
题目分析:
这是一道可以用DP算法来解决的题目,当然用暴力破解也没问题,固定一个字符串中的一个特定字符,然后遍历另一个字符串来找相同子串也可以解决,不过缺乏美感,而且时间复杂度高,达到O(n^3)。用DP来解,转换方程设置为if (A[i] == B[j]) A[i][j] = A[i - 1][j - 1] + 1,这样只要记录下A[i][j](0 <= i < A.size(), 0 <= j < B.size())的最大值,就能得到相同子串的最大长度,只用O(n^2)的时间复杂度。
具体步骤:
- step1:设置二维数组记录到某个点为止的最大共同字串长度,并初始化为全0。
- step2:遍历所有A和B的所有字符,如果遇到相同字符,则改变当前位置长度值数组值。
- step3:第一行和第一列的遇到相同字符直接设置为1,其它行和列的找上一个位置记录下的最大长度加1。
- step4:假如此时的最大长度比之前记录的最大值大,则替换为当前当前相同子串的长度值。
- step5:返回所有位置的相同子串最大长度值max。
DP解法代码:
#include <iostream>
using namespace std;
#include <vector>
#include <string>
/*DP算法关键是找到初始状态方程和状态转移方程,代码并不复杂*/
class Solution {
public:
int findLength(vector<int>& A, vector<int>& B) {
int max = 0;
/*设置二维数组记录到某个点为止的最大共同字串长度,并初始化为全0*/
vector<vector<int> > f(A.size(), vector<int>(B.size(), 0));
for (int i = 0; i < A.size(); i++){
for (int j = 0; j < B.size(); j++){
if (A[i] == B[j]){ //遇到相同字符时,长度要发生改变
if (i == 0 || j == 0) f[i][j] = 1; //第一行和第一列的遇到相同字符直接设置为1
else f[i][j] = f[i - 1][j - 1] + 1; //其它行和列的找上一个位置记录下的最大长度加1
if (f[i][j] > max) max = f[i][j]; //假如此时的最大长度比之前记录的最大值大,则替换为当前当前相同子串的长度值
}
}
}
return max;
}
};