【力扣每日一题】组合数|||——偷鸡写法(python)

本文介绍了一个编程问题,如何找到所有使用数字1到9且不超过k次,其和等于n的组合。利用Python的itertools库中的combinations函数暴力求解,并给出了一个Solution类的实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

216. 组合总和 III

找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:

  • 只使用数字1到9
  • 每个数字 最多使用一次 

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。

示例 3:

输入: k = 4, n = 1
输出: []
解释: 不存在有效的组合。
在[1,9]范围内使用4个不同的数字,我们可以得到的最小和是1+2+3+4 = 10,因为10 > 1,没有有效的组合。

关于组合问题,选和不选,枚举选哪个都是比较经典的问题,但是python的强大之一就是库的丰富,我们可以用itertools库中的combinations函数来解决组合问题,比较暴力,仅供参考。

class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        q = [i for i in range(1,10)]
        w = []
        for i in combinations(q, k):
            if sum(i) == n:
                w.append(i)
        return w

from itertools import combinations

# 生成指定长度的所有可能的组合
# combinations(iterable, r):返回iterable中所有长度为r的子序列,返回的子序列中的项按输入iterable中的顺序排序
iterable = ['a', 'b', 'c']
combs = list(combinations(iterable, 2))

print(combs)  # 输出结果为 [('a', 'b'), ('a', 'c'), ('b', 'c')]
 

 

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值