题目:
给定三个字符串 s1、s2、s3,请你帮忙验证 s3 是否是由 s1 和 s2 交错 组成的。
两个字符串 s 和 t 交错 的定义与过程如下,其中每个字符串都会被分割成若干 非空 子字符串:
s = s1 + s2 + ... + sn
t = t1 + t2 + ... + tm
|n - m| <= 1
交错 是 s1 + t1 + s2 + t2 + s3 + t3 + ... 或者 t1 + s1 + t2 + s2 + t3 + s3 + ...
提示:a + b 意味着字符串 a 和 b 连接。
思路
未进行滚动数组优化的代码:
class Solution {
public:
bool isInterleave(string s1, string s2, string s3) {
int m=s1.length();
int n=s2.length();
bool f[m+1][n+1];
if(m+n!=s3.length()){
return false;
}
memset(f,false,sizeof(f));
f[0][0]=true;
for(int i=0;i<=m;i++){
for(int j=0;j<=n;j++){
int now = i+j-1;
if(i==0&&j==0)
continue;
else if(i==0){
if((s2[j-1]==s3[now])&&f[i][j-1]==true)
f[i][j]=true;
}
else if(j==0){
if((s1[i-1]==s3[now])&&f[i-1][j]==true)
f[i][j]=true;
}
else{
if(f[i-1][j]==true&&(s1[i-1]==s3[now])){
f[i][j]=true;
}
if(f[i][j-1]==true&&(s2[j-1]==s3[now])){
f[i][j]=true;
}
}
}
}
return f[m][n];
}
};
优化后的代码:
class Solution {
public:
bool isInterleave(string s1, string s2, string s3) {
auto f = vector <int> (s2.size() + 1, false);
int n = s1.size(), m = s2.size(), t = s3.size();
if (n + m != t) {
return false;
}
f[0] = true;
for (int i = 0; i <= n; ++i) {
for (int j = 0; j <= m; ++j) {
int p = i + j - 1;
if (i > 0) {
//与之前的f[j]进行&操作,就相当于与上一行同一列的f[j]进行&操作。
f[j] &= (s1[i - 1] == s3[p]);=
}
if (j > 0) {
//将同一行前面一列取得值与自身进行或操作。
f[j] |= (f[j - 1] && s2[j - 1] == s3[p]);
}
}
}
return f[m];
}
};