LeetCode Scramble String

LeetCode解题之Scramble String

原题

一个字符串可以拆分成两个都不为空的子字符串,而子字符串(长度大于等于二)也可以不断这样拆分下去,现在可以任意交换拆分出来两部分的位置来改变字符串中字符的顺序。判断两个字符能否通过这种方式相互转换。

注:这道题比较难用语言描述,可以参见原题中的图例

原题请点 这里

注意点:

  • 给的两个字符串的长度相等

例子:

输入: s1 = “rgtae”, s2 = “great”

输出: True (“rgtae”->”grtae”->”greta”->”great”)

解题思路

对三维动态规划还不是很熟练,偷懒用了最简单的递归方式,以后会补上动态规划解法。要判断两个字符S和T能否转化,先把它们各自分为两部分,如果S的前半部分和T的前半部分能转换,它们的后半部分也能转换,说明它们就能转换;但也有可能S的前半部分和后半部分是在最后一交换中转换回来的,也就是S的前半部分和T的后半部分能够转换,而T的前半部分和S的后半部分能够转换同样能够达到目的。还可以在我代码的基础上再做一些优化,如提前判断两个要转化的字符串中各字符的数目是否相等来进行剪枝,来减少没有用的递归。经过剪枝的递归算法的运行速度还是很快的。

AC源码

from collections import defaultdict


class Solution(object):
    def isScramble(self, s1, s2):
        """ :type s1: str :type s2: str :rtype: bool """
        if s1 == s2:
            return True
        count1 = defaultdict(int)
        count2 = defaultdict(int)
        for e1, e2 in zip(s1, s2):
            count1[e1] += 1
            count2[e2] += 1
        if count1 != count2:
            return False
        for i in range(1, len(s1)):
            if self.isScramble(s1[:i], s2[:i]) and self.isScramble(s1[i:], s2[i:]) \
                    or self.isScramble(s1[:i], s2[-i:]) and self.isScramble(s1[i:], s2[:len(s2) - i]):
                return True
        return False

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。

你可能感兴趣的:(LeetCode,算法,python,递归,动态规划)