文巾解题 784. 字母大小写全排列

Posted 2021-09-06 UQI-LIUWJ

1 题目描述

 2 解题思路：回溯

        如果解决一个问题有多个步骤，每一个步骤有多种方法，题目又要我们找出所有的方法，可以使用回溯算法；
        回溯算法是在一棵树上的 深度优先遍历（因为要找所有的解，所以需要遍历）；
        组合问题，相对于排列问题而言，不计较一个组合内元素的顺序性（即 [1, 2, 3] 与 [1, 3, 2] 认为是同一个组合），因此很多时候需要按某种顺序展开搜索，这样才能做到不重不漏。

class Solution(object):
    def letterCasePermutation(self, S):
        ret=[]

        def dfs(lst,tmp):
            if(len(tmp)==1):
                if(tmp[0]>='0' and tmp[0]<='9'):
                    lst1=lst.copy()
                    lst1.append(tmp[0])
                    ret.append(lst1)
                else:
                    lst1=lst.copy()
                    lst1.append(tmp[0].upper())
                    ret.append(lst1)

                    lst2=lst.copy()
                    lst2.append(tmp[0].lower())
                    ret.append(lst2)
            else:
                if(tmp[0]>='0' and tmp[0]<='9'):
                    lst1=lst.copy()
                    lst1.append(tmp[0])
                    dfs(lst1,tmp[1:])
                else:
                    lst1=lst.copy()
                    lst1.append(tmp[0].upper())
                    dfs(lst1,tmp[1:])

                    lst2=lst.copy()
                    lst2.append(tmp[0].lower())
                    dfs(lst2,tmp[1:])
'''
dfs(lst,tmp)表示，我当前已经操作完成的字符串在lst，剩余还没有操作的在tmp中
如果当前tmp只有一个元素的，那么就把排列结果放入ret数组中（根据这个元素是数字还是字母再分类讨论一次）
否则，根据tmp当前位是数字还是字母，分类讨论向后递归
'''        

        dfs([],S)
        res=[]

        for i in ret:
            res.append(''.join(i))
#ret部分是一个二维数组，需要join一下
        return(res)