题目:
给两个整数数组 nums1
和 nums2
,返回 两个数组中 公共的 、长度最长的子数组的长度 。
示例一:
输入:nums1 = [1,2,3,2,1], nums2 = [3,2,1,4,7] 输出:3
示例二:
输入:nums1 = [0,0,0,0,0], nums2 = [0,0,0,0,0] 输出:5
提示:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 100
思路:
这道题使用动态规划中的dp数组来组,首先想要考虑的问题是dp数组的含义是什么。
这道题dp数组的含义为nums1[i]和nums2[j]位置对应的最长公共子数组。
接下来先看下面的这幅图
我们将两个数组组合成为一个二维数组来计算,下一步我们要考虑的计算dp数组的初始化。
这里我使用了将第一行和第一列分别初始化
for(int i = 0; i < row; i++){//赋值第一列
if(nums1[i] == nums2[0]){
dp[i][0] = 1;
ans = max(ans, dp[i][0]);
}
}
for(int i = 0; i < col; i++){//赋值第一行
if(nums1[0] == nums2[i]){
dp[0][i] = 1;
ans = max(ans, dp[0][i]);
}
}
赋值第一列时,我们将nums1的每一个数去和nums2[0]去做比较,若相等,则dp[i][0]赋值为1,反之赋值为0.
同理在赋值第一行时将nums2的每一个数去和nums1[0]去做比较。
赋值结束之后我们就要进行dp数组后面的计算了,若此时nums1[i] == nums2[j],则
dp[i][j] = dp[i - 1][j - 1] + 1;
全部完成之后我们就要考虑返回值的问题,我们有两种方法,第一种就是去使用for循环去遍历dp数组去找到最大值,第二种就是在赋值以及计算时去使用max函数来记录下最大值,最后返回最大值。
代码:
class Solution {
public:
int findLength(vector<int>& nums1, vector<int>& nums2) {
//dp[i][j]代表是nums1[i]和nums2[j]对应的最长公共子数组
int row = nums1.size();//行数
int col = nums2.size();//列数
int ans = 0;
vector<vector<int>> dp(row, vector<int>(col));
for(int i = 0; i < row; i++){//赋值第一列
if(nums1[i] == nums2[0]){
dp[i][0] = 1;
ans = max(ans, dp[i][0]);
}
}
for(int i = 0; i < col; i++){//赋值第一行
if(nums1[0] == nums2[i]){
dp[0][i] = 1;
ans = max(ans, dp[0][i]);
}
}
for(int i = 1; i < row; i++){
for(int j = 1; j < col; j++){
if(nums1[i] == nums2[j]){
dp[i][j] = dp[i - 1][j - 1] + 1;//相等的话就是前一个位置的最长公共子数组加一
ans = max(ans, dp[i][j]);
}
}
}
return ans;
}
};
时间复杂度:O(n * m)
n为nums1的长度,m为num2的长度
空间复杂度:O(n * m)
n为nums1的长度,m为num2的长度