Leetcode_med 77. 组合

原创于 2019-03-21 11:12:11 发布 · 181 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

75 篇文章

订阅专栏

博客主要描述了一个问题，即给定两个整数n和k，要返回1到n中所有可能的k个数的组合。并提到根据题型应使用backtracking遍历，还表示会直接使用Python库来解决该问题。

描述

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

Python

看题型知道应该使用backtracking遍历

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        res = []
        self.dfs(range(1,n+1),k,0,[],res)
        return res
    def dfs(self,nums,k,index,path,res):
        if k==0:
            res.append(path)
            return
        for i in range(index,len(nums)):
            self.dfs(nums,k-1,i+1,path+[nums[i]],res)

直接上python库

from itertools import combinations

class Solution:
    def combine(self, n, k):
        return list(combinations(range(1, n+1), k))