给定一个字符串S
,通过将字符串S
中的每个字母转变大小写,我们可以获得一个新的字符串。返回所有可能得到的字符串集合。
示例:
输入: S = "a1b2"
输出: ["a1b2", "a1B2", "A1b2", "A1B2"]
输入: S = "3z4"
输出: ["3z4", "3Z4"]
输入: S = "12345"
输出: ["12345"]
注意:
S
的长度不超过12
。S
仅由数字和字母组成。
第一种思路:
看到题目要求组合或者集合,马上想到可以用回溯法:回溯法本来是说对于每个元素都先考虑放它的情况,再考虑不放它的情况;放在这道题的背景里就是,对于每个字母,先考虑放它,再考虑放它的另一种大小写形式。
用dfs实现回溯,start代表目前从扫描到第几位,
如果是digit,就直接加进去,然后下一层递归
如果是alpha,就先加进去,然后下一层递归;再加对立大小写形式, 然后下一层递归。
class Solution(object):
def letterCasePermutation(self, S):
"""
:type S: str
:rtype: List[str]
"""
res = list()
l = len(S)
if l == 0:
return [""]
def dfs(start, temp):
if start >= l or len(temp) == l:
res.append(temp)
return
# print start, temp
if S[start].isdigit():
dfs(start + 1, temp + S[start])
elif S[start].islower():
dfs(start + 1, temp + S[start])
dfs(start + 1, temp + S[start].upper())
elif S[start].isupper():
dfs(start + 1, temp + S[start])
dfs(start + 1, temp + S[start].lower())
dfs(0, "")
return res
#以下是修改前
class Solution(object):
def letterCasePermutation(self, S):
"""
:type S: str
:rtype: List[str]
"""
res = list()
l = len(S)
if l == 0:
return [""]
def dfs(string, start, temp):
if start > l or len(temp) == l:
res.append(temp)
return
if string[start].isdigit():
temp += string[start]
dfs(string, start + 1, temp)
if string[start].islower():
temp += (string[start])
dfs(string, start + 1, temp)
temp = temp[:-1]
temp += (string[start].upper())
dfs(string, start + 1, temp)
elif string[start].isupper():
temp += string[start]
dfs(string, start + 1, temp)
temp = temp[:-1]
temp += string[start].lower()
dfs(string, start + 1, temp)
dfs(S, 0, "")
return res
第二种思路:
除了用DFS回溯实现,我们也可以用BFS来解题, 线性扫描S,
对于扫描到的每个元素, 都把它的大小写形式分别加到,目前的res里的所有结果里,这样可以得到temp,
然后用temp覆盖res。
比如对于S = "a1b2",
扫描到a时, res = [a, A]
扫描到b时, res = [a1, A1], temp = [a1b, a1B, A1b, A1B]
class Solution(object):
def letterCasePermutation(self, S):
"""
:type S: str
:rtype: List[str]
"""
import copy
res = [""]
for i, x in enumerate(S):
if x.isdigit():
for index, item in enumerate(res):
res[index] += (x)
elif x.isupper():
temp = list()
for index, item in enumerate(res):
temp.append(item + (x))
temp.append(item + (x.lower()))
res = copy.deepcopy(temp[:])
elif x.islower():
temp = list()
for index, item in enumerate(res):
temp.append(item + (x))
temp.append(item + (x.upper()))
res = copy.deepcopy(temp[:])
return res
第三种思路:
Bitmap法,字符串S的长度为l, 则总共会有 2** l种结果,换成二进制就是0 ~ 2 **l - 1个数,
对于每个数,如果某个位上是0, 就放小写;是1, 就放大写。
class Solution(object):
def letterCasePermutation(self, S):
l = len(S)
n = 2 ** l
res = list()
if l == 0:
res.append("")
for i in range(0, n):
temp = ""
for j in range(0, l):
if ((2 ** j) &i) == 0:
temp += S[j].lower()
else:
temp += S[j].upper()
if temp not in res:
res.append(temp)
return res
本题跟LeetCode-Python-78. 子集的思路非常类似。