题目:1035. 不相交的线
方法一:递归,由顶向下。状态sta[i][j]表示:数组nums1区间[0,i]与数组nums2区间[0,j]之间可以匹配的最大数。
1、当nums1[i]==nums2[j]时,说明答案可以有sta[i][j]=sta[i-1][j-1]+1
2、相等或不相等时,答案也可以为sta[i][j]=max(sta[i][j],max(sta[i-1][j],sta[i][j-1]))。
细节看注释。
class Solution {
public:
int dfs(int i,int j,vector<int>& nums1, vector<int>& nums2,vector<vector<int>>& sta){
//当有一个小于0时,说明到达边界了,直接返回0即可
if(i<0||j<0) return 0;
//之前出现并计算过,直接返回
if(sta[i][j]!=-1) return sta[i][j];
int res=0;
//相等时
if(nums1[i]==nums2[j]) res=dfs(i-1,j-1,nums1,nums2,sta)+1;
res=max(res,max(dfs(i-1,j,nums1,nums2,sta),dfs(i,j-1,nums1,nums2,sta)));
//记忆化
return sta[i][j]=res;
}
int maxUncrossedLines(vector<int>& nums1, vector<int>& nums2) {
int n=nums1.size(),m=nums2.size();
//由于递归是自顶向下,会有很多重复的状态,所以采用数组来存储进行记忆化搜索
vector<vector<int>> sta(n,vector<int>(m,-1));
return dfs(n-1,m-1,nums1,nums2,sta);
}
};
方法二:递推,由底向上。状态f[i][j]表示:数组nums1区间[0,i]与数组nums2区间[0,j]之间可以匹配的最大数。这里+1是为了更方便处理边界情况。
class Solution {
public:
int maxUncrossedLines(vector<int>& nums1, vector<int>& nums2) {
int n=nums1.size(),m=nums2.size();
vector<vector<int>> f(n+1,vector<int>(m+1,0));
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(nums1[i]==nums2[j]) f[i+1][j+1]=f[i][j]+1;
f[i+1][j+1]=max(f[i+1][j+1],max(f[i][j+1],f[i+1][j]));
}
}
return f[n][m];
}
};
方法三:递推空间优化,观察上面的代码,会发现,其实只用到了当前行,以及上一行。我们可以用记录两行来替代记录n行。但更简便的是,用pre来记录上一行的前一个数即可。
class Solution {
public:
int maxUncrossedLines(vector<int>& nums1, vector<int>& nums2) {
int n=nums1.size(),m=nums2.size();
vector<int> f(m+1,0);
for(int i=0;i<n;i++){
//用pre来记录上一行的前一个数
int pre=0;
for(int j=0;j<m;j++){
int tmp=f[j+1];
if(nums1[i]==nums2[j]) f[j+1]=pre+1;
f[j+1]=max(f[j+1],max(tmp,f[j]));
pre=tmp;
}
}
return f[m];
}
};