Leetcode 87. 扰乱字符串 Java/Python

给定一个字符串 s1，我们可以把它递归地分割成两个非空子字符串，从而将其表示为二叉树。

下图是字符串 s1 = "great" 的一种可能的表示形式。

    great
   /    \
  gr    eat
 / \    /  \
g   r  e   at
           / \
          a   t

在扰乱这个字符串的过程中，我们可以挑选任何一个非叶节点，然后交换它的两个子节点。

例如，如果我们挑选非叶节点 "gr" ，交换它的两个子节点，将会产生扰乱字符串 "rgeat" 。

    rgeat
   /    \
  rg    eat
 / \    /  \
r   g  e   at
           / \
          a   t

我们将 "rgeat” 称作 "great" 的一个扰乱字符串。

同样地，如果我们继续将其节点 "eat" 和 "at" 进行交换，将会产生另一个新的扰乱字符串 "rgtae" 。

    rgtae
   /    \
  rg    tae
 / \    /  \
r   g  ta  e
       / \
      t   a

我们将 "rgtae” 称作 "great" 的一个扰乱字符串。

给出两个长度相等的字符串 s1 和 s2，判断 s2 是否是 s1 的扰乱字符串。

示例 1:

输入: s1 = "great", s2 = "rgeat"
输出: true

示例 2:

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

思路：使用递归，要把字符串分成两个部分，s1分成a1，a2，s1分成b1，b2，判断（a1-b2) && (a2-b2) 和（a1-b2）&& (a2-b1)

java代码：

class Solution {
    
    private int[] frequency(char[] sa,int from,int to){
        int[] f = new int[26];
        for(int i = from; i <= to;i++ ){
            f[sa[i] - 'a']++ ;
        }
        return f;
    }
    
    private boolean equalFrequency(int[] f1,int[] f2){
        for (int i = 0;i < f1.length;i++){
            if (f1[i] != f2[i]) return false;
        }
        return true;
    }
    
    private boolean scramble(char[] sa1,int f1,int t1,char[] sa2,int f2,int t2){
        if (f1- t1 != f2-t2) return false;
        if (f1> t1) return false;
        if (f1 == t1) return sa1[f1] == sa2[f2];
        //对比两个字符是否相等不管顺序
        int[] c1 = frequency(sa1,f1,t1);
        int[] c2 = frequency(sa2,f2,t2);
        if (!equalFrequency(c1,c2)) return false;
        for(int i = 0 ;i < t1-f1; i++){
            if (scramble(sa1,f1,f1+i,sa2,f2,f2+i) && scramble(sa1,f1+i+1,t1,sa2,f2+i+1,t2)) return true;
            if (scramble(sa1,f1,f1+i,sa2,t2-i,t2) && scramble(sa1,f1+i+1,t1,sa2,f2,t2-i-1)) return true;
        }
        return false;
    }
    
    
    public boolean isScramble(String s1, String s2) {
        char[] sa1 = s1.toCharArray();
        char[] sa2 = s2.toCharArray();
        return scramble(sa1,0,sa1.length-1,sa2,0,sa2.length-1);
    }
}

Python:

class Solution(object):
    def isScramble(self, s1, s2):
        """
        :type s1: str
        :type s2: str
        :rtype: bool
        """
        if len(s1)!=len(s2): 
            return False
        if s1==s2: 
            return True
        l1=list(s1)
        l2=list(s2)
        l1.sort()
        l2.sort()
        if l1!=l2: 
            return False
        length=len(s1)
        for i in range(1,length):
            if self.isScramble(s1[:i],s2[:i]) and self.isScramble(s1[i:],s2[i:]): 
                return True
            if self.isScramble(s1[:i],s2[length-i:]) and self.isScramble(s1[i:],s2[:length-i]): 
                return True
        return False