代码随想录算法训练营第二十七天|39.组合总和、40.组合总和II、131.分割回文
39.组合总和
给你一个 无重复元素 的整数数组 candidates
和一个目标整数 target
,找出 candidates
中可以使数字和为目标数 target
的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates
中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target
的不同组合数少于 150
个。
示例 1:
输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。
题解:套用回溯模板。需要注意的是,数组中的元素可以重复使用,那么for循环中的回溯参数需要是i,表示当前元素可以选取。剪枝的话,如果路径中的和已经大于目标值,就直接返回。
代码:
class Solution {
List<List<Integer>> res=new ArrayList<>();
List<Integer> path=new ArrayList<>();
public List<List<Integer>> combinationSum(int[] candidates, int target) {
backtracking(candidates,target,0,0);
return res;
}
//回溯算法
public void backtracking(int [] candi,int target,int sum,int startIndex){
if(sum>target) return;
if(sum==target){
res.add(new ArrayList<>(path));
}
//单层逻辑
for(int i=startIndex;i<candi.length;i++){
path.add(candi[i]);
sum+=candi[i];
backtracking(candi,target,sum,i);
path.removeLast();
sum-=candi[i];
}
}
}
40.组合总和II
给定一个候选人编号的集合 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]
]
题解:回溯算法中的去重问题。怎么去重呢?它是横向不能有重复的,可是竖向可以有重复的,并且每个元素只能用一次,所以用一个boolean类型的used数组,将遍历过的元素的下标置为true。并且将数组排序,如果当前元素和前一个元素相等,并且前一个元素已经遍历过了的话(used数组下标为true),就跳过这个值往下遍历。
代码:
class Solution {
List<List<Integer>> res=new ArrayList<>();
List<Integer> path=new ArrayList<>();
boolean [] used;
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
used=new boolean[candidates.length];
Arrays.fill(used,false);
Arrays.sort(candidates);
backtracking(candidates,target,0,0);
return res;
}
public void backtracking(int [] candidates,int target,int sum,int startIndex){
if(sum>target) return;
if(sum==target){
res.add(new ArrayList<>(path));
return;
}
//去重逻辑的实现
for(int i=startIndex;i<candidates.length;i++){
//跳过搜索
if(i>0 && candidates[i]==candidates[i-1] && !used[i-1]){
continue;
}
used[i]=true;
path.add(candidates[i]);
sum+=candidates[i];
backtracking(candidates,target,sum,i+1);
sum-=candidates[i];
path.removeLast();
used[i]=false;
}
}
}
131.分割回文
给你一个字符串 s
,请你将 s
分割成一些子串,使每个子串都是
回文串
。返回 s
所有可能的分割方案。
示例 1:
输入:s = "aab"
输出:[["a","a","b"],["aa","b"]]
示例 2:
输入:s = "a"
输出:[["a"]]
题解:判断回文的方法:双指针,分别从前和从后开始扫描。切割问题,就是选取元素的过程,只是将选取的元素组成一个字串,并且是回文字串的情况下才加入到path集合中。需要注意的是 substring()方法是前闭后开的。
代码:
class Solution {
List<List<String>> res=new ArrayList<>();
List<String> path=new ArrayList<>();
public List<List<String>> partition(String s) {
backtracking(s,0);
return res;
}
//双指针法判断回文
public boolean isHui(String s,int start,int end){
while(start<end){
if(s.charAt(start)!=s.charAt(end))
return false;
start++;
end--;
}
return true;
}
public void backtracking(String s,int startIndex){
//说明找到一个回文子串(竖着下来已经遍历完了)
if(startIndex==s.length()){
res.add(new ArrayList<>(path));
return;
}
for(int i=startIndex;i<s.length();i++){
if(isHui(s,startIndex,i)){
String str=s.substring(startIndex,i+1);
path.add(str);
backtracking(s,i+1);
path.removeLast();
}else{
continue;
}
}
}
}