扰乱字符串-算法题

题意：给定两个字符串，问能其中一个字符串是否是经过另一个字符串“扰乱”后得到的。
扰乱的定义：若s1=x+y，s2=y+x，则s1是s2扰乱后的结果。其中扰乱可以一直递归下去。即a+b+c+d = d+c+b+a = d+c+a+b。
传送门

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

题解：DFS，通过下标id1和id2以及长度length，判断s1[id1…id1+length]与s2[id2…id2+length]是否满足条件。然后暴力选取中间点进行深搜。

bool isScramble(string s1, string s2) {
    if (s1.length() != s2.length()) return false;
    
    function<bool(int,int,int)> checkIfSimilar = [&](int i1, int i2, int length) {
        unordered_map<int, int> freq;
        for (int i = i1; i < i1+length; ++i)
            ++freq[s1[i]];
        for (int i = i2; i < i2+length; ++i) {
            --freq[s2[i]];
            if (!freq[s2[i]]) freq.erase(s2[i]);
        }
        if (freq.empty()) return true;
        return false;
    };

    int n = s1.length();
    vector<vector<vector<int>>> dp(n, vector<vector<int>>(n, vector<int>(n+1, -1)));

    function<int(int,int,int)> dfs = [&](int i1, int i2, int length) {
        if (dp[i1][i2][length] != -1)
            return dp[i1][i2][length];

        if (s1.substr(i1, length) == s2.substr(i2, length))
            return dp[i1][i2][length] = 1;

        if (!checkIfSimilar(i1, i2, length))
            return dp[i1][i2][length] = 0;
            
        for (int i = 1; i < length; ++i) {
            if (dfs(i1, i2, i) && dfs(i1+i, i2+i, length-i))
                return dp[i1][i2][length] = 1;

            if (dfs(i1, i2+length-i, i) && dfs(i1+i, i2, length-i))
                return dp[i1][i2][length] = 1;
        }
        return dp[i1][i2][length] = 0;
    };

    return dfs(0, 0, n);
}