0216.-组合总和 III

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

所有数字都是正整数。
解集不能包含重复的组合。 
示例 1:

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii

参考:

python

# 0216.组合总和III

class Solution:
    def combinationSum3(self, k: int, n: int) -> [[int]]:
        res = []
        path = []

        def findFinalPath(n,k,sum,startIndex):
            if sum > n:
                return # 剪枝操作
            if sum == n and len(path) == k:
                return res.append(path[:])
            # 不剪枝的for loop
            # for i in range(startIndex, 10):
            for i in range(startIndex, 9-(k-len(path))+2):
                path.append(i)
                sum += i
                findFinalPath(n,k,sum,i+1) # 调整startIndex
                sum -= i
                path.pop()

        findFinalPath(n,k,0,1)
        return res

golang

package backTrack


// 回溯
func combinationSum3(k,n int) [][]int {
	var path []int
	var res [][]int
	backTrackSum3(n,k,1,&path,&res)
	return res
}

func backTrackSum3(n,k,startIndex int, path *[]int, res *[][]int)  {
	// 递归终止条件
	if len(*path) == k {
		var sum int
		tmp := make([]int, k)
		for k,v := range *path {
			sum += v
			tmp[k] = v
		}
		if sum == n {
			*res = append(*res, tmp)
		}
		return // 即使 len==k, 但sum!=n也return跳出
	}
	// i < 9-(k-len(*path))+1做了剪枝,(k-len(*track)表示还剩多少个可填充的元素),不剪枝则 i < 10(1-9)
	for i:=startIndex;i<=9-(k-len(*path))+1;i++ {
		// 记录path
		*path = append(*path, i)
		// 递归
		backTrackSum3(n,k,i+1,path,res)
		// 回溯
		*path = (*path)[:len(*path)-1]
	}
}

posted on 2021-11-21 21:07  进击的davis  阅读(24)  评论(0编辑  收藏  举报

导航