Leetcode之回溯法专题-39. 组合总数(Combination Sum)
Leetcode之回溯法专题-39. 组合总数(Combination Sum)
给定一个无重复元素的数组 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]
]
分析:给定一个目标值,给定N个数字,要从N个数字里取数字(可以重复选取)
求这些数字的和等于目标值的所有情况。
用回溯法很好解决这个问题,首先从0-(n-1)中选取第1个数,第一个数取1-k次,以此类推。
AC代码如下:
class Solution { List<List<Integer>> ans = new ArrayList<>(); public List<List<Integer>> combinationSum(int[] candidates, int target) { dfs(candidates,target,0,target,new ArrayList<Integer>()); return ans; } public void dfs(int[] candidates,int target,int index,int balance,ArrayList<Integer> list){ if(balance<0){ return; } if(balance==0){ ans.add(new ArrayList<Integer>(list)); return; } if(index>=candidates.length){ return; } int num = candidates[index]; int n = balance/num +1; for(int i=0;i<=n;i++){ for(int j=0;j<i;j++){ list.add(num); } int count = num*i; dfs(candidates,target,index+1,balance-count,list); for(int j=0;j<i;j++){ list.remove(list.size()-1); } } } }
作者:秦羽纶
本文版权归作者所有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文链接,否则保留追究法律责任的权利.