leetcode 之 Combinations

描述:

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],
]
即 排列组合的问题. 对于例子来说,就是C4, 2, 从4个里面取出两个来.
之前有做过全排列的问题和求所有子集的问题. 见:   http://www.cnblogs.com/missmzt/p/5527382.html

这里的解决方法都是通过回溯法来解决. 个人对于回溯法还不是很熟练,主要在于回溯时条件函数的设计吧. 
例如之前的 全排列问题是 通过一个标记数组, 标记第i个数是否放入结果集合中.
对于组合问题, 当回溯判断 集合的长度达到k时, 这时应该返回上一层, 开始回溯. 在回溯的过程中,我们设定开始的数和结束的数.
代码如下:

class Solution(object):
    def combine(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: List[List[int]]
        """
        
        res = []
        fin = []
        self.helper(res, 0, n, k, fin)
        return fin
        
    def helper(self, res, s, n, k, fin):
        if len(res) == k:
            fin.append(list(res))   # python的深拷贝与浅拷贝
            return
        for i in range(s, n):
            res.append(i+1)
            self.helper(res, i+1, n, k, fin)  
            res.pop()

运行结果:　　[[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4]]

求子集的代码如下：

#!/usr/bin/env python
#coding=utf-8

def ziji(cur, n, a, b):
    if cur == n:
        b = []
        for i in range (0, n):
            if a[i] == 1:
                b.append(i+1)
        print b
        return

    a[cur] = 1
    ziji(cur+1, n, a, b)
    a[cur] = 0
    ziji(cur+1, n, a, b)
a =  [0]*4
ziji(0, 4, a, []);