2023-02-12 Day25 回溯算法

17. 电话号码的字母组合

给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母
题目链接:17. 电话号码的字母组合 - 力扣(Leetcode)

itertools.product太强大了,可以将列表中的元素组合出所有的顺序,而且是根据列表中元素的个数进行组合(笛卡儿积)

class Solution:
    def letterCombinations(self, digits: str) -> List[str]:
        import itertools
        phonenum = {'2':'abc','3':'edf','4':'ghi','5':'jkl','6':'mno','7':'pqrs','8':'tuv','9':'wxyz'}
        if not digits:
            return list()
        group = [phonenum[digit] for digit in digits]
        res = [''.join(c) for c in itertools.product(*group)]
        return res 

示例:

输入:digits = "23"
输出:["ad","ae","af","bd","be","bf","cd","ce","cf"]

itertools.product(group)的星号应该是展开为单独的元素,即group中每个元素取出一个,然后product.
在 Python 中,星号(
)有多种用法,具体取决于它所出现的上下文。
当在函数参数中使用时,* 用于指示函数可以接受任意数量的位置参数。例如:

def foo(*args):
    for arg in args:
        print(arg)
foo(1, 2, 3)
这将输出:
1
2
3

在这个例子中,*args 中的 * 告诉 Python 接受任意数量的位置参数,并将它们收集到一个名为 args 的元组中。

当在函数调用中使用时,* 用于将可迭代对象展开为单独的位置参数。例如:

def bar(a, b, c):
    print(a, b, c)

args = [1, 2, 3]
bar(*args)
这将输出:
1 2 3

在这个例子中,*args 中的 * 告诉 Python 将 args 列表的元素展开为单独的位置参数,并将它们传递给 bar 函数。

当在列表或元组中使用时,* 用于将列表或元组的元素展开为单独的元素。例如:

a = [1, 2, 3]
b = [4, 5, 6]
c = [*a, *b]
print(c)
这将输出:
[1, 2, 3, 4, 5, 6]

在这个例子中,[*a, *b] 中的 * 告诉 Python 将 a 和 b 列表的元素展开为单独的元素。

216. 组合总和 III - 力扣(Leetcode)

找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回

class Solution:
    def __init__(self):
        self.res=[]
        self.sum=0
        self.path=[]
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        self.backtracing(k,n,1)
        return self.res 
    def backtracing(self,k,n,startindex):
        if self.sum>n:
            return
        if len(self.path)==k:
            if self.sum==n:
                self.res.append(self.path[:])
            return 
        for i in range(startindex,10-(k-len(self.path))+1):
            self.path.append(i)
            self.sum +=i
            self.backtracing(k,n,i+1)
            self.path.pop()
            self.sum -=i 
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容