Leetcode 77. Combinations 组合

Leetcode 77. Combinations 组合

标签 ： Leetcode

---

题目地址： https://leetcode-cn.com/problems/combinations/

题目描述

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

示例:

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

算法思想

这个主要利用回溯法进行处理，我们用current数组保存当前遍历到的，如果len(current) ==k,那就说明当前已经够了，否则，则把当前的节点加入到current里面，外面的for循环的起始是在原有基础上加一，说明了之前遍历过的不能遍历，并且保证没有重复比如1 3和3 1.

python代码

class Solution(object):
    def dfs(self,i,n,k,res,current):
        if len(current) == k:
            res.append(copy.copy((current)))
        else:
            for tmp in range(i,n+1):
                current.append(tmp)
                self.dfs(tmp+1,n,k,res,current)
                current.pop()

    def combine(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: List[List[int]]
        """
        res = []
        self.dfs(1,n,k,res,[])
        return res