难度:困难
使用下面描述的算法可以扰乱字符串 s 得到字符串 t :
- 如果字符串的长度为 1 ,算法停止
- 如果字符串的长度 > 1 ,执行下述步骤:
- 在一个随机下标处将字符串分割成两个非空的子字符串。即,如果已知字符串
s, 则可以将其分成两个子字符串x和y,且满足s = x + y。 - 随机 决定是要「交换两个子字符串」还是要「保持这两个子字符串的顺序不变」
。即,在执行这一步骤之后,
s可能是s = x + y或者s = y + x。 - 在
x和y这两个子字符串上继续从步骤 1 开始递归执行此算法。
- 在一个随机下标处将字符串分割成两个非空的子字符串。即,如果已知字符串
给你两个 长度相等 的字符串 s1 和 s2,判断 s2 是否是 s1 的扰乱字符串
。如果是,返回 true ;否则,返回 false 。
输入:s1 = "great", s2 = "rgeat"
输出:true
解释:s1 上可能发生的一种情形是:
"great" --> "gr/eat" // 在一个随机下标处分割得到两个子字符串
"gr/eat" --> "gr/eat" // 随机决定:「保持这两个子字符串的顺序不变」
"gr/eat" --> "g/r / e/at" // 在子字符串上递归执行此算法。两个子字符串分别在随机下标处进行一轮分割
"g/r / e/at" --> "r/g / e/at" // 随机决定:第一组「交换两个子字符串」,第二组「保持这两个子字符串的顺序不变」
"r/g / e/at" --> "r/g / e/ a/t" // 继续递归执行此算法,将 "at" 分割得到 "a/t"
"r/g / e/ a/t" --> "r/g / e/ a/t" // 随机决定:「保持这两个子字符串的顺序不变」
算法终止,结果字符串和 s2 相同,都是 "rgeat"
这是一种能够扰乱 s1 得到 s2 的情形,可以认为 s2 是 s1 的扰乱字符串,返回 true
输入:s1 = "abcde", s2 = "caebd"
输出:false
输入:s1 = "a", s2 = "a"
输出:true
显然「扰乱字符串」的关系是具有对称性的,即如果s1是s2的扰乱字符串,那么s2也
是s1的扰乱字符串,为了叙述方便,我们称这种情况下,s1和s2是「和谐」的。
那么如何判断s1和s2是和谐的?我们首先可以想到几个简单的判断方法:
- 如果
s1=s2,那么他们是和谐的; - 如果
s1和s2长度不同,那么它们一定不是和谐的; - 如果
s1中某个字符c出现了x1次,而c在s2中出现了x2次,且x1不等 于x2,那么它们一定不是和谐的。
那么对于剩下的情况,我们可以从s1的分割方法入手。假设s1作为根节点时被分割
成l(s1)和r(s1)两个子串,那么:
- 如果
l(s1)和r(s2)没有被交换,那么s2需要存一种分割方 式s2 = l(s2) + r(s2),使得l(s1)和l(s2)是和谐的,并且r(s1)和r(s2)是和 谐的; - 如果
l(s1)和r(s2)被交换了,那么s2需要存在一种分割方 法s2 = l(s2) + r(s2),使得l(s1)和r(s2)是和谐的,并且r(s1)和l(s2)是和 谐的
这样一来,我们就把原本要解决的问题划分为两个本质相同、但规模更小的子问题,因此可 以考虑使用动态规划解决。
设f(s1,s2)来表示s1和s2是否和谐,那么可以写出状态转移:
- 当
s1 = s2时,f(s1,s2)为 true, - 当存在某个字符
c,它在s1和s2中出现的次数不同,则f(s1, s2)为 false
因为题目保证给定的原始字符串的长度相同,因此我们只需要判断上面两种情况,如
果s1和s2不符合这两种情况,那么我们需要枚举分割点。
设s1和s2的长度为n,我们用s1(x, y)表示s1从第x个字符(从 0 开始编号)
开始,长度为y的子串。由于分割出的两个字符串不能为空串,那么其中一个字符串就
是s1(0, i),另一个字符串为s1(i, n-i)。
- 对于
l(s1)和r(s1)没有被交换的情况,s2同样需要被分为s2(0,i)以 及s2(i, n-i),否则长度不同的字符串是不可能和谐的,因此我们可以写出状态转移:- 当
i从1到n-1遍历,但凡有一个时候满足f(s1(0, i), s2(0, i))为 true,同 时f(s1(i,n-i), s2(i,n-i))为 true 时,则代表f(s1, s2)为 true,否则为 false。
- 当
- 对于
l(s1)和r(s1)被交换的情况,s2需要被分为s2(0,n-i)以及s2(n-i, i), 这样对应的长度才会相同。因此我们可以写出状态转移:- 当
i从1到n-1遍历,但凡有一个时候满足f(s1(0, i), s2(n-i, i))为 true, 同时f(s1(i,n-i), s2(0,n-i))为 true 时,则代表f(s1, s2)为 true,否则为 false。
- 当
因此我们可以实现我们的代码:
/**
* 动态规划
* @desc 时间复杂度 O(N^4) 空间复杂度 O(N^3)
* @param s1
* @param s2
*/
export function isScramble(s1: string, s2: string): boolean {
const len = s1.length;
const memo: number[][][] = new Array(len)
.fill([])
.map(() => new Array(len).fill([]).map(() => new Array(len + 1).fill(0)));
return dfs(0, 0, len, s1, s2, memo);
function dfs(
i1: number,
i2: number,
len: number,
s1: string,
s2: string,
memo: number[][][]
): boolean {
if (memo[i1][i2][len] !== 0) {
return memo[i1][i2][len] === 1;
}
// 判断两个子串是否相等
if (s1.slice(i1, i1 + len) === s2.slice(i2, i2 + len)) {
memo[i1][i2][len] = 1;
return true;
}
// 判断是否存在字符 c 在两个子串中出现的次数不同
if (!checkIfSimilar(i1, i2, len, s1, s2)) {
memo[i1][i2][len] = -1;
return false;
}
// 枚举分割位置
for (let i = 1; i < len; i++) {
// 不交换的情况
if (
dfs(i1, i2, i, s1, s2, memo) &&
dfs(i1 + i, i2 + i, len - i, s1, s2, memo)
) {
memo[i1][i2][len] = 1;
return true;
}
// 交换的情况
if (
dfs(i1, i2 + len - i, i, s1, s2, memo) &&
dfs(i1 + i, i2, len - i, s1, s2, memo)
) {
memo[i1][i2][len] = 1;
return true;
}
}
memo[i1][i2][len] = -1;
return false;
}
function checkIfSimilar(
i1: number,
i2: number,
len: number,
s1: string,
s2: string
): boolean {
const freq = new Map<string, number>();
for (let i = i1; i < i1 + len; i++) {
const c = s1[i];
freq.set(c, (freq.get(c) || 0) + 1);
}
for (let i = i2; i < i2 + len; i++) {
const c = s2[i];
freq.set(c, (freq.get(c) || 0) - 1);
}
for (const value of freq.values()) {
if (value !== 0) {
return false;
}
}
return true;
}
}