vlambda博客
学习文章列表

力扣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; }};