力扣718——最长重复子数组(动态规划)
祝大家新年快乐,思路就简单写了写
题目(中等)
给两个整数数组 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
解题思路
dp[i][j]表示以i,j结尾的连续的公共子数组长度;
初始值全是0;
转移方程为,如果nums1[i] == nums2[j],dp[i][j] = dp[i-1][j-1] + 1;
注意边界处理;
其中最大值即为答案。
代码
class Solution {
public:
int findLength(vector<int>& nums1, vector<int>& nums2) {
int n1 = nums1.size(), n2 = nums2.size();
vector<vector<int>> dp(n1, vector<int>(n2, 0));
int ans = 0;
for(int i = 0; i < n1; i++) {
for(int j = 0; j < n2; j++) {
if(nums1[i] == nums2[j]) {
dp[i][j] = (i > 0 && j > 0) ? dp[i-1][j-1] + 1 : 1;
ans = max(ans, dp[i][j]);
}
else dp[i][j] = 0;
}
}
return ans;
}
};