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) 来获得相关源代码。