Skip to content

Latest commit

 

History

History
195 lines (160 loc) · 6.81 KB

README.md

File metadata and controls

195 lines (160 loc) · 6.81 KB

扰乱字符串

难度:困难

https://leetcode-cn.com/problems/scramble-string/

题目

使用下面描述的算法可以扰乱字符串 s 得到字符串 t

  1. 如果字符串的长度为 1 ,算法停止
  2. 如果字符串的长度 > 1 ,执行下述步骤:
    • 在一个随机下标处将字符串分割成两个非空的子字符串。即,如果已知字符串 s , 则可以将其分成两个子字符串 xy ,且满足 s = x + y
    • 随机 决定是要「交换两个子字符串」还是要「保持这两个子字符串的顺序不变」 。即,在执行这一步骤之后,s 可能是 s = x + y 或者 s = y + x
    • xy 这两个子字符串上继续从步骤 1 开始递归执行此算法。

给你两个 长度相等 的字符串 s1s2,判断 s2 是否是 s1 的扰乱字符串 。如果是,返回 true ;否则,返回 false 。

示例

示例 1:

输入: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

示例 2:

输入:s1 = "abcde", s2 = "caebd"
输出:false

示例 3:

输入:s1 = "a", s2 = "a"
输出:true

解题

动态规划

显然「扰乱字符串」的关系是具有对称性的,即如果s1s2的扰乱字符串,那么s2也 是s1的扰乱字符串,为了叙述方便,我们称这种情况下,s1s2是「和谐」的。

那么如何判断s1s2是和谐的?我们首先可以想到几个简单的判断方法:

  • 如果s1 = s2,那么他们是和谐的;
  • 如果s1s2长度不同,那么它们一定不是和谐的;
  • 如果s1中某个字符c出现了x1次,而cs2中出现了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)是和 谐的

scramble-string

这样一来,我们就把原本要解决的问题划分为两个本质相同、但规模更小的子问题,因此可 以考虑使用动态规划解决。

f(s1,s2)来表示s1s2是否和谐,那么可以写出状态转移:

  • s1 = s2 时,f(s1,s2)为 true,
  • 当存在某个字符c,它在s1s2中出现的次数不同,则f(s1, s2)为 false

因为题目保证给定的原始字符串的长度相同,因此我们只需要判断上面两种情况,如 果s1s2不符合这两种情况,那么我们需要枚举分割点。

s1s2的长度为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),否则长度不同的字符串是不可能和谐的,因此我们可以写出状态转移:
    • i1n-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), 这样对应的长度才会相同。因此我们可以写出状态转移:
    • i1n-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;
  }
}