今天是第 天刷leetcode,立个flag,打卡60天,如果做不到,完成一件评论区点赞最高的挑战。
算法挑战链接
216. 组合总和 IIIhttps://leetcode.cn/problems/combination-sum-iii/description/
第一想法
这道题和第二十一天做的题目很像,不过这道题目添加了一个条件,需要列表中的值等于N。但是我们都可以遍历所有的可能了,那再加上一个计算是否等于N那不是简简单单,于是用昨天的代码写出来的代码如下:
class Solution {
LinkedList<Integer> result1 = new LinkedList<>();
List<List<Integer>> results1 = new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
backtracking1(n, k, 1);
return results1;
}
void backtracking1(int n, int k,int index){
if (result1.size() == k && result1.stream().mapToInt(i->i).sum() == n) {
results1.add(new ArrayList<>(result1));
return;
}
for (int i = index; i <= 9; i++){
result1.add(i);
backtracking1(n, k, i+1);
result1.removeLast();
}
}
}
条件判断出添加了 result1.stream().mapToInt(i->i).sum() == n 。
看完代码随想录之后的想法
看完代码随想录之后,发现暴力的回溯会遍历所有的可能,但是有一些可能是在开始或者中间的时候是可以判断出这条路径是走不通的,于是可以进行剪枝。在这道题目中,如果在进入递归的时候,列表中的值的和大于N了,那就没有必要往下回溯了。
代码如下:
class Solution {
LinkedList<Integer> result1 = new LinkedList<>();
List<List<Integer>> results1 = new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
backtracking1(n, k, 1);
return results1;
}
void backtracking1(int n, int k,int index){
int sum = result1.stream().mapToInt(i->i).sum();
if (sum > n) return;
if (result1.size() == k && sum == n) {
results1.add(new ArrayList<>(result1));
return;
}
for (int i = index; i <= 9; i++){
result1.add(i);
backtracking1(n, k, i+1);
result1.removeLast();
}
}
}
实现过程中遇到哪些困难
剪枝要如何思考?
剪枝就是将不必要的路径移除,那如何判断该路径是否必要呢?这道题目中是sum>n 就没有必要了,