码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode 77, Combinations

时间:2016-12-09 08:41:56      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:type   ...   元素   lin   搜索   nat   ranch   div   n+1   

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的搜索。没有这两行的话,程序超时。

 

Leetcode 77, Combinations

标签:type   ...   元素   lin   搜索   nat   ranch   div   n+1   

原文地址:http://www.cnblogs.com/lettuan/p/6148021.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!