问题描述
题目要求: 输入一个正整数型的数组和一个正整数target,找出用数组中元素组成的所有组合,使得组合的和等于target。注意,数组中元素可以重复使用
解题思路
很经典的递归题目,每次使用需要记录一些辅助的值,所以重新写一个backtrack函数用来辅助操作。传入的参数是当前需要检验的集合,数组中使用的数据开始位置start以及target的余额,每次检查到当前的target为0时说明当前的待检验集合是一个解加入到结果集中。
代码实现
public List<List<Integer>> combinationSum(int[] candidates, int target) {
ArrayList res = new ArrayList<List<Integer>>();
backtrack(res, new ArrayList<Integer>(), candidates, target, 0);
return res;
}
public void backtrack(List<List<Integer>> list, ArrayList<Integer> tmplist, int[] nums, int target, int start) {
//由于所有元素都是整数,所以当target余额不足的时候直接返回即可
if(target < 0) return;
else if(target == 0) {
//当target余额为零说明当前的集合是一个解
list.add(new ArrayList(tmplist));
}
else {
for(int i = start; i < nums.length; i++) {
tmplist.add(nums[i]);
//注意,由于需要考虑同一个元素多次使用,所以这里start的位置还是原来的地方
backtrack(list, tmplist, nums, target-nums[i], i);
tmplist.remove(tmplist.size() - 1);
}
}
}
复杂度分析:
假设数组的长度是n,target的值是m,worst case的情况是数组中所有的元素都是1,target只能检测很多次的1,1,1,1,1…1这样,如果用树来理解的话,树的深度是m,每一层有n个节点,所以时间复杂度是O(nm),空间复杂度是O(n*k),其中k代表符合条件的解的个数