39.组合总和

题目描述

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明:

所有数字(包括 target)都是正整数。
解集不能包含重复的组合。 

示例1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

示例2:

输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]

思路

1.这道题可以使用回溯算法求解。
2.记录当前元素的位置,以及在结果集中加入当前元素之后的target数值。
3.若target<0则说明无解,要进行回溯操作,跳回上一个元素的位置。
4.若target=0则说明该组合是目标解之一,将其加入结果集之后,进行回溯操作。

Java代码实现

  public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> ans = new ArrayList<>();
        backtrack(candidates,target,0,res,ans);
        return res;
    }

  private void backtrack(int[] candidates,int target,int start,List<List<Integer>> res,List<Integer> ans){
        if(target<0)
            return;
        if(target == 0){
            res.add(new ArrayList<>(ans));
            return;
        }
        for (int i = start; i < candidates.length; i++) {
            ans.add(candidates[i]);
            backtrack(candidates,target-candidates[i],i,res,ans);
            ans.remove(ans.size()-1);
        }
    }
func combinationSum(candidates []int, target int) [][]int {
    res := make([][]int,0)
    combinationSumDFS(candidates,0,target,&res,[]int{})
    return res
}

func combinationSumDFS(candidates []int,curIndex int, target int,res *[][]int,cur []int){
    if target < 0 {
        return
    }
    if target == 0 {
        copyCur := make([]int,len(cur))
        copy(copyCur,cur)
        *res = append(*res, copyCur)
        return
    }
    for i:=curIndex; i<len(candidates);i++ {
        cur = append(cur, candidates[i])
        combinationSumDFS(candidates,i,target-candidates[i],res,cur)
        cur = cur[:len(cur)-1]
    }
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务