leetcode40 组合总和II

题目

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意:解集不能包含重复的组合。

示例

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:[[1,1,6],[1,2,5],[1,7],[2,6]]

解析

本题中每个数字在每个组合中可以使用多次;但最后的组合不能有相同的;
本题中原始数据中有重复元素;
也就是说,树的深度不用去重,宽度的时候需要去重;
而上一道题,是原始数据中没有重复元素,但一个元素可以用多次;
注意:这种的剪枝都需要排序

func combinationSum2(candidates []int, target int) [][]int {
    var path []int
    var res [][]int
    sort.Ints(candidates) //也是需要排序
    backtracking(0, 0, target, candidates, path, &res)
    return res
}

func backtracking(startIndex, sum, target int, candidates, path []int, res *[][]int) {
    if sum == target {
        tmp := make([]int, len(path))
        copy(tmp, path)
        *res = append(*res, tmp)
        return
    }
    if sum > target {
        return
    }

    for i := startIndex; i < len(candidates) && sum + candidates[i] <= target; i++ {
        // 如果当前树层有使用过相同元素,则continue
        if i > startIndex && candidates[i] == candidates[i-1] {
            continue // 注意是continue,不是return
        }
        path = append(path, candidates[i])
        sum += candidates[i]
        backtracking(i+1, sum, target, candidates, path, res) //这里是i+1,因为每个数字在每个组合中只能用一次
        path = path[:len(path)-1]
        sum -= candidates[i]
    }
}

你可能感兴趣的:(#,leetcode回溯系列,算法,数据结构)