Scramble String
Given a string s1, we may represent it as a binary tree by partitioning it to two non-empty substrings recursively.
Below is one possible representation of s1 = “great”:
Given two strings s1 and s2 of the same length, determine if s2 is a scrambled string of s1.
Example 1:
Input: s1 = “great”, s2 = “rgeat”
Output: true
Example 2:
Input: s1 = “abcde”, s2 = “caebd”
Output: false
解法1:递归
如果s1和s2是 scramble 的话,那么必然存在一个在 s1 上的长度 l1,将 s1 分成 s11 和 s12 两段,同样有 s21 和 s22,那么要么 s11 和 s21 是 scramble 的并且 s12 和 s22 是 scramble 的;要么 s11 和 s22 是 scramble 的并且 s12 和 s21 是 scramble 的。
class Solution {
public:
bool isScramble(string s1, string s2) {
if(s1.size() != s2.size()) return false;
if(s1 == s2) return true;
string str1 = s1, str2 = s2;
sort(str1.begin(), str1.end());
sort(str2.begin(), str2.end());
if(str1 != str2) return false;
for(int i=1;i<str1.size();i++){
string s11 = s1.substr(0,i);
string s12 = s1.substr(i);
string s21 = s2.substr(0, i);
string s22 = s2.substr(i);
if (isScramble(s11, s21) && isScramble(s12, s22)) return true;
s21 = s2.substr(s1.size() - i);
s22 = s2.substr(0, s1.size() - i);
if (isScramble(s11, s21) && isScramble(s12, s22)) return true;
}
return false;
}
};
解法2:动态规划
维护一个三维数组dp[i][j][n],i是s1的起始字符,j是s2的起始字符,n是字符串长度
dp[i][j][n]表示s1以i开始,s2以j开始的长度为n的字符串是不是scramble。
判断这个是不是满足,其实我们首先是把当前 s1[i…i+len-1] 字符串劈一刀分成两部分,然后分两种情况:
第一种是左边和 s2[j…j+len-1] 左边部分是不是 scramble,以及右边和 s2[j…j+len-1] 右边部分是不是 scramble;
第二种情况是左边和 s2[j…j+len-1] 右边部分是不是 scramble,以及右边和 s2[j…j+len-1] 左边部分是不是 scramble。
如果以上两种情况有一种成立,说明 s1[i…i+len-1] 和 s2[j…j+len-1] 是 scramble 的。
而对于判断这些左右部分是不是 scramble 我们是有历史信息的,因为长度小于n的所有情况我们都在前面求解过了(也就是长度是最外层循环)。
上面说的是劈一刀的情况,对于 s1[i…i+len-1] 我们有 len-1 种劈法,在这些劈法中只要有一种成立,那么两个串就是 scramble 的。
总结起来递推式是 res[i][j][len] = || (res[i][j][k] && res[i+k][j+k][len-k] || res[i][j+len-k][k]&&res[i+k][j][len-k]) 对于所有 1<=k<len
class Solution {
public:
bool isScramble(string s1, string s2) {
if(s1.size() != s2.size()) return false;
if(s1 == s2) return true;
int n = s1.size();
vector<vector<vector<bool> > > dp (n, vector<vector<bool> >(n, vector<bool>(n + 1, false)));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
dp[i][j][1] = (s1[i] == s2[j]);
for(int len=2;len<=n;len++)
for(int i=0;i<=n-len;i++)
for(int j=0;j<=n-len;j++)
for(int k=1;k<len;k++)
if((dp[i][j][k] && dp[i+k][j+k][len-k]) || (dp[i + k][j][len - k] && dp[i][j + len - k][k]))
dp[i][j][len] = true;
return dp[0][0][n];
}
};