美文网首页算法提高之LeetCode刷题算法
784. 字母大小写全排列(Python)

784. 字母大小写全排列(Python)

作者: 玖月晴 | 来源:发表于2019-05-28 10:01 被阅读0次

    题目

    难度:★★☆☆☆
    类型:字符串

    给定一个字符串S,通过将字符串S中的每个字母转变大小写,我们可以获得一个新的字符串。返回所有可能得到的字符串集合。

    注意
    S 的长度不超过12。
    S 仅由数字和字母组成。

    示例

    输入: S = "a1b2"
    输出: ["a1b2", "a1B2", "A1b2", "A1B2"]

    输入: S = "3z4"
    输出: ["3z4", "3Z4"]

    输入: S = "12345"
    输出: ["12345"]

    解答

    这里参考celia评论

    方案1:回溯法

    回溯法(探索与回溯法)是一种选优搜索法,又称为试探法,按选优条件向前搜索,以达到目标。但当探索到某一步时,发现原先选择并不优或达不到目标,就退回一步重新选择,这种走不通就退回再走的技术为回溯法,而满足回溯条件的某个状态的点称为“回溯点”。(来自百度百科)。

    在这道题中,回溯法从左到右扫描字符串,当遇到字母时考虑使用该字母的大写和小写形式两种可能,直到遍历到字符串最后一个字符。

    这里我们定义函数dfs(start, tmp)实现回溯,函数的输入有两个:start是遍历到的当前位,tmp是遍历到当前位时已经记录的前半部分子串。回溯的过程:针对当前位S[start](该字符赋给了变量c)的不同进行不同操作:

    1. 首先判断当前是否到达回溯终点,这里根据tmp的长度是否恰好为输入字符串S的长度判断;

    2. 如果没有到达回溯终点,则判断该字符c的类型:
      (1)如果是数字,则直接进行下一轮回溯:dfs(start+1, tmp+c);
      (2)如果是大写字母,则考虑使用它本身以及使用它的小写形式两种情况:dfs(start+1, tmp+c)和dfs(start+1, start+c.lower());
      (3)如果是小写字母,则考虑使用它本身以及使用它的大写形式两种请款:dfs(start+1, tmp+c)和dfs(start+1, start+c.upper())。

    这里注意输入为空的处理。

    class Solution(object):
        def letterCasePermutation(self, S):
    
            res = []
    
            if not S:
                return [""]
    
            def dfs(start, tmp):
                if len(tmp) == len(S):  # 已经找到了一个答案
                    res.append(tmp)
                    return
                
                c = S[start]
                # print start, temp
                if c.isdigit():  # 数字就直接加
                    dfs(start + 1, tmp + c)
                elif S[start].islower():
                    # 字母就加本身和对立面
                    dfs(start + 1, tmp + c)
                    dfs(start + 1, tmp + c.upper())
                else:
                    dfs(start + 1, tmp + c)
                    dfs(start + 1, tmp + c.lower())
    
            dfs(0, "")
            return res
    

    有些逻辑可以合并:

    class Solution(object):
        def letterCasePermutation(self, S):
    
            if not S:
                return [""]
    
            res = []
    
            def dfs(start, temp):
                if len(temp) == len(S):  # 已经找到了一个答案
                    res.append(temp)
                    return
    
                c = S[start]
                dfs(start + 1, temp + c)
                if not c.isdigit():
                    c = c.upper() if c.islower() else c.lower()
                    dfs(start + 1, temp + c)
    
            dfs(0, "")
            return res
    

    方案2:线性扫描

    我们可以直接使用线性方式进行扫描,每次扫描从结果列表中取出临时子串,并根据当前字符的情况在临时子串后添加字符:
    (1)如果当前字符是数字:直接添加;
    (2)如果当前字符是大写字母:除添加本身外,还要添加小写形式;
    (3)如果当前字符是小写字母:除添加本身外,还要添加大写形式。

    每次用新的结果列表更新原列表。

    class Solution(object):
        def letterCasePermutation(self, S):
    
            res = [""]
    
            for c in S:
                next_res = []
                for tmp in res:
                    next_res.append(tmp+c)
                    if not c.isdigit():
                        c = c.upper() if c.islower() else c.lower()
                        next_res.append(tmp + c)
                res = next_res
    
            return res
    

    如有疑问或建议,欢迎评论区留言~

    相关文章

      网友评论

        本文标题:784. 字母大小写全排列(Python)

        本文链接:https://www.haomeiwen.com/subject/ajqktctx.html