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

lintcode_152.组合

时间:2018-01-05 12:22:48      阅读:113      评论:0      收藏:0      [点我收藏+]

标签:回溯思想   整数   结果   gpo   递归   ber   log   ntc   元素   

组给出两个整数n和k,返回从1......n中选出的k个数的组合。

样例

例如 n = 4 且 k = 2

返回的解为:

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

 

递归回溯思想,用k控制内循环次数,从而控制每一个元素中数的个数

class Solution:
    """    
    @param n: Given the range of numbers
    @param k: Given the numbers of combinations
    @return: All the combinations of k numbers out of 1..n   
    """
    def combine(self, n, k):      
        # write your code here  
        self.res = []
        tmp = []
        self.dsf(n,k,1,0,tmp)
        
        return self.res
        
    def dsf(self, n, k, m, p, tmp):
        if k == p: #当p == k:说明tmp中元素个数为k,可以添加到结果中
            self.res.append(tmp[:])
            return
        for i in range(m, n+1):
            tmp.append(i)
            self.dsf(n, k, i+1, p+1, tmp) #p也代表了递归次数
            tmp.pop() #将最后一位剔除,更新新的元素

 

 

lintcode_152.组合

标签:回溯思想   整数   结果   gpo   递归   ber   log   ntc   元素   

原文地址:https://www.cnblogs.com/zhangli-ncu/p/8203225.html

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