Leetcode 77, Combinations

最新推荐文章于 2020-06-17 13:19:51 发布

转载最新推荐文章于 2020-06-17 13:19:51 发布 · 51 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/lettuan/p/6148021.html

本文介绍了一个经典的组合生成问题：从1到n的整数中选取k个数字的所有可能组合，并提供了一种递归深度优先搜索（DFS）算法来解决这个问题。通过具体的Python代码示例，展示了如何高效地生成这些组合，避免不必要的重复计算。

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

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

 1 class Solution(object):
 2     def combine(self, n, k):
 3         """
 4         :type n: int
 5         :type k: int
 6         :rtype: List[List[int]]
 7         """
 8         res = []
 9         nums = list(range(1, n+1))
10         self.dfs(nums, k, 0, res, [])
11         
12         return res
13         
14     def dfs(self, nums, k, step, res, line):
15         if len(nums) < k - step:
16             return
17         if step == k:
18             res.append([x for x in line])
19         
20         for i, x in enumerate(nums):
21             line.append(nums[i])
22             self.dfs(nums[i+1:], k, step + 1, res, line)
23             line.pop() 
24

L15-L16 判段一下剩下的可以填的元素个数是不是比剩下的空位少，如果是的话可以提前结束这个branch的搜索。没有这两行的话，程序超时。L22说明，如果去了一个num之后，在line里面这个num之后的数只取比它大的数，这样就保证了最后结果中不会出现重复的情况，比如在答案中只可能出现[1,2,3]，而不可能出现[2,1,3]。

转载于:https://www.cnblogs.com/lettuan/p/6148021.html