题目描述
题目链接:87. Scramble String
We can scramble a string s to get a string t using the following algorithm:
- If the length of the string is 1, stop.
- If the length of the string is > 1, do the following:
- Split the string into 2 non-empty substrings at a random index, i.e. if the string is
s
, divide it tox
and y wheres = x + y
.- Randomly, decide to swap the two substrings or to keep them in the same order. i.e. after this step,
s
may becomes = x + y
ors = y + x
.- Apply step 1 recursively on each of the two substrings
x
andy
.Given two strings
s1
ands2
of the same length, return true ifs2
is a scrambled string ofs1
, otherwise, return false.
例子
例子 1
Input: s1 = “great”, s2 = “rgeat” Output: true Explanation:One possible scenario applied on s1 is: “great” –> “gr/eat” // divide at random index. “gr/eat” –> “gr/eat” // random decision is not to swap the two substrings and keep them in order. “gr/eat” –> “g/r / e/at” // apply the same algorith recursively on both substrings. divide at ranom index each of them. “g/r / e/at” –> “r/g / e/at” // random decision was to swap the first substing and to keep the second substring in the same order. “r/g / e/at” –> “r/g / e/ a/t” // again apply the algorithm recursively, divide “at” to “a/t”. “r/g / e/ a/t” –> “r/g / e/ a/t” // random decision is to keep both substings in the same order. The algorithm stops now and the result string is “rgeat” which is s2. As there is one possible scenario that led s1 to be scrambled to s2, we return true.
例子 2
Input: s1 = “abcde”, s2 = “caebd” Output: false
例子 3
Input: s1 = “a”, s2 = “a” Output: true
Constraints
s1.length == s2.length
1 <= s1.length <= 30
s1
ands2
consist of lower-case English letters.
解题思路
记忆化 + 递归搜索
首先注意到题目给出的测试集规模字符串最长在 30, 而且只有小写字母组成。因此通过 记忆化+递归搜索 结合剪枝是可以通过的。具体思路是:
- 初始化一个二维哈希表
record
,记录每一个s1
和s2
的匹配结果 - 如果
record[s1][s2]
存在,直接返回对应结果 (剪枝 1) - 当给定
s1
和s2
长度为 1,直接返回s1 == s2
- 当
s1 == s2
,返回true
- 利用哈希表统计
s1
和s2
的字符频率是否一致,如果不一致直接返回false
(剪枝 2,这一步可以减少很多分支) - 长度从
1
至s1.length() - 1
,依次分割 s1 为前后两部分(s1_1, s1_2
):(以下对s2
的子串均是根据对应s1
的长度来分割,不改变顺序)- a. 递归调用
Sramble(s1_1, s2_1)
记录其结果在record[s1_1][s2_1]
,如果record[s1_1][s2_1] == false
跳过 b (剪枝 3) - b. 递归调用
Sramble(s1_2, s2_2)
记录在record[s1_2][s2_2]
, 如果record[s1_2][s2_2] == true
返回true
重新进入 5 (剪枝 4) - c. 重复 a, b 步骤,不过将
s1_1
和s1_2
调换前后顺序
- a. 递归调用
- 当上述循环结束均没有返回时,返回
false
代码如下所示:
#include <string>
#include <vector>
#include <unordered_map>
class Solution {
public:
bool isScramble(std::string s1, std::string s2) {
return Scramble(s1, s2);
}
private:
std::unordered_map<std::string, std::unordered_map<std::string, bool>> record;
bool Scramble(const std::string& s1, const std::string& s2) {
if (record.find(s1) != record.end() && record[s1].find(s2) != record[s1].end()) return record[s1][s2];
if (s1.length() == 1) {
return s1 == s2;
} else if (s1 == s2) {
return true;
}
std::vector<int> count(26, 0);
for (char letter: s1) {
count[letter - 'a']++;
}
for (char letter: s2) {
if (count[letter - 'a'] == 0) {
record[s1][s2] = false;
}
count[letter - 'a']--;
}
for (int i = 1; i < s1.length(); i++) {
std::string s1_sub1 = s1.substr(0, i);
std::string s1_sub2 = s1.substr(i, s1.length() - i);
record[s1_sub1][s2.substr(0, i)] = Scramble(s1_sub1, s2.substr(0, i));
if (record[s1_sub1][s2.substr(0, i)]) {
record[s1_sub2][s2.substr(i, s1.length() - i)] = Scramble(s1_sub2, s2.substr(i, s1.length() - i));
if (record[s1_sub2][s2.substr(i, s1.length() - i)]) {
return true;
}
}
record[s1_sub2][s2.substr(0, s1.length() - i)] = Scramble(s1_sub2, s2.substr(0, s1.length() - i));
if (record[s1_sub2][s2.substr(0, s1.length() - i)]) {
record[s1_sub1][s2.substr(s1.length() - i, i)] = Scramble(s1_sub1, s2.substr(s1.length() - i, i));
if (record[s1_sub1][s2.substr(s1.length() - i, i)]) {
return true;
}
}
}
return false;
}
};
- 时间复杂度: O(2^n)? 不确定
- 空间复杂度: O(n^4) 二维哈希表所需空间