元宵节快乐~昨天因为有点事没有写,今天补上。继续回溯算法的练习。
开始做题!
39.组合总和(中等)
给你一个 无重复元素 的整数数组 candidates
和一个目标整数 target
,找出 candidates
中可以使数字和为目标数 target
的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates
中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target
的不同组合数少于 150
个。
示例 1:
输入:candidates =[2,3,6,7],
target =7
输出:[[2,2,3],[7]]
思路:本题跟之前做的组合总和问题的区别主要就是在于本题的同一个 数字可以 无限制重复被选取 。那么就需要在之前的代码基础上对startindex做出改变。
代码实现:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
class Solution {
List<List<Integer>>result=new ArrayList<>();
LinkedList<Integer>path=new LinkedList<>();
public List<List<Integer>> combinationSum(int[] candidates, int target) {
Arrays.stream(candidates).sorted();
backtracking(candidates,target,0,0);
return result;
}
public void backtracking(int[]candidates,int target,int sum,int index){
if(sum==target){
result.add(new ArrayList<>(path));
return;
}
if(sum>target)return;
for (int i = index; i < candidates.length; i++) {
if(i>0)
if(candidates[i]<candidates[i-1])return;
path.add(candidates[i]);
sum+=candidates[i];
backtracking(candidates,target,sum,i);
path.removeLast();
sum-=candidates[i];
}
}
}
总结:回溯算法中,方法函数中的参数最好都得用index。遇到可以重复使用数字的问题可以借鉴此题。
40、组合总和2(中等)
给定一个候选人编号的集合 candidates
和一个目标数 target
,找出 candidates
中所有可以使数字和为 target
的组合。
candidates
中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。
示例 1:
输入: candidates =[10,1,2,7,6,1,5]
, target =8
, 输出: [ [1,1,6], [1,2,5], [1,7], [2,6] ]
思路:通过画图,去重操作可以根据这个写出代码。
如果纵向到了第二轮选择(i>index)且树的左右重复(candidates[i] == candidates[i - 1]),就跳过。
代码实现:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
class Solution {
List<List<Integer>>result=new ArrayList<>();
LinkedList<Integer>path=new LinkedList<>();
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
Arrays.sort(candidates);
backtracking(candidates,target,0,0);
return result;
}
public void backtracking(int[]candidates,int target,int sum,int index){
if(sum==target){
result.add(new ArrayList<>(path));
return;
}
if(sum>target)return;
for (int i = index; i < candidates.length; i++) {
if (i>0)
if (i > index && candidates[i] == candidates[i - 1]) {
continue;
}
path.add(candidates[i]);
sum+=candidates[i];
backtracking(candidates,target,sum,i+1);
path.removeLast();
sum-=candidates[i];
}
}
}
总结:掌握去重的操作,注意画图写代码。