87. 扰乱字符串(动态规划)

package com.heu.wsq.leetcode.dp;

import java.util.HashMap;
import java.util.Map;

/**
 * 87. 扰乱字符串
 * @author wsq
 * @date 2021/4/16
 * 使用下面描述的算法可以扰乱字符串 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 。
 *
 * 示例 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
 *
 * 链接：https://leetcode-cn.com/problems/scramble-string
 */
public class IsScramble {
    int[][][] memo;
    String s1, s2;

    /**
     * 动态规划
     * memo[i][j][len] 表示从s1的i位置起，从s2的j位置起，长度为len的字符串是否是扰乱字符串
     * 扰乱字符串可以分为两类：
     * 1. 按照某个位置分割后的x,y子串不需要更换两者的位置
     * 2. 按照某个位置分割后的x,y子串需要更换两者的位置
     * 分割的位置可以是[1,len)，由于字符串的长度最大为30，因此可以枚举每一个位置。（如果字符串长度过长，可能会出现超时的情况）
     * @param s1
     * @param s2
     * @return
     */
    public boolean isScramble(String s1, String s2){
        int len = s1.length();
        this.s1 = s1;
        this.s2 = s2;
        memo = new int[len][len][len+1];
        return dfs(0, 0, len);
    }

    private boolean dfs(int i1, int i2, int len){
        if(memo[i1][i2][len] != 0){
            return memo[i1][i2][len] == 1;
        }
        // 判断两个子串是否相等
        if (s1.substring(i1, i1 + len).equals(s2.substring(i2, i2 + len))){
            memo[i1][i2][len] = 1;
            return true;
        }
        // 判断存在字符c在两个子串中的数量是否相同
        if(!checkIfSimilar(i1, i2, len)){
            memo[i1][i2][len] = -1;
            return false;
        }
        // 其他情况，枚举分割位置
        for (int i = 1; i < len; i++){
            // 不交换的情况
            if (dfs(i1, i2, i) && dfs(i1 + i, i2 + i, len - i)){
                memo[i1][i2][len] = 1;
                return true;
            }
            // 交换的情况
            if (dfs(i1, i2 + len - i, i) && dfs(i1 + i, i2, len - i)){
                memo[i1][i2][len] = 1;
                return true;
            }
        }
        memo[i1][i2][len] = -1;
        return false;
    }

    private boolean checkIfSimilar(int i1, int i2, int len){
        Map<Character, Integer> map = new HashMap<>();
        for (int i = 0; i < len; i++){
            char c = this.s1.charAt(i1 + i);
            map.put(c, map.getOrDefault(c, 0) + 1);
        }

        // 遍历s2，判断s1中的字符数量与s2的是否相等
        for (int j = 0; j < len; j++){
            char c = this.s2.charAt(i2 + j);
            map.put(c, map.getOrDefault(c, 0) - 1);
            if(map.get(c) < 0){
                return false;
            }
        }

        // 遍历map，查看每个字符对应的数量是否为0
        for(Map.Entry<Character, Integer> entry: map.entrySet()){
            if(entry.getValue() != 0){
                return false;
            }
        }
        return true;
    }
}