LeetCode-Python-1087. 字母切换

我们用一个特殊的字符串 S 来表示一份单词列表，之所以能展开成为一个列表，是因为这个字符串 S 中存在一个叫做「选项」的概念：

单词中的每个字母可能只有一个选项或存在多个备选项。如果只有一个选项，那么该字母按原样表示。

如果存在多个选项，就会以花括号包裹来表示这些选项（使它们与其他字母分隔开），例如 "{a,b,c}" 表示 ["a", "b", "c"]。

例子："{a,b,c}d{e,f}" 可以表示单词列表 ["ade", "adf", "bde", "bdf", "cde", "cdf"]。

请你按字典顺序，返回所有以这种方式形成的单词。

 

示例 1：

输入："{a,b}c{d,e}f"
输出：["acdf","acef","bcdf","bcef"]

示例 2：

输入："abcd"
输出：["abcd"]

 

提示：

1. 1 <= S.length <= 50
2. 你可以假设题目中不存在嵌套的花括号
3. 在一对连续的花括号（开花括号与闭花括号）之间的所有字母都不会相同

思路：

问不同组合，上回溯。

如果扫描到单个字符，那么直接加到当前答案tmp的末尾，然后继续往后找。

如果扫描到{，说明需要把{}里夹着的元素都找出来，把它们分别加到当前答案tmp的末尾，然后继续往后找。

坑点在输出需要按字典序，所以每个花括号里的元素要排序。

class Solution(object):
    def permute(self, S):
        """
        :type S: str
        :rtype: List[str]
        """
        def generate(s, tmp):
            if not s:
                res.append(tmp[:])
                return
            if s[0].isalpha():
                generate(s[1:], tmp + s[0])
            else:
                for i in range(len(s)):
                    if s[i] == "}":
                        substring = s[1:i]
                        break
                chars = substring.split(",")
                chars.sort()
                for char in chars:
                    generate(s[i + 1:], tmp + char)
        res = []
        generate(S, "")
        
        return res