数据结构刷题(十九):77组合、216组合总和III
创始人
2024-05-30 01:29:52

1.组合

题目链接

  1. 过程图:先从集合中取一个数,再依次从剩余数中取k-1个数。

  1. 思路:回溯算法。使用回溯三部曲进行解题:

  • 递归函数的返回值以及参数:n,k,startIndex(记录每次循环集合从哪里开始遍历的位置),其中startIndex 就是防止出现重复的组合。比如从1开始了循环,则使用startindex=2,让startindex作为下次循环的开始。

还有全局变量:一个是用来存放一个符合条件的结果path,一个用来存放所有符合条件的结果集合result。

  • 回溯函数终止条件:path这个数组的大小如果达到k,说明我们找到了一个子集大小为k的组合,在图中path存的就是根节点到叶子节点的路径

  • 单层搜索的过程:for循环用来横向遍历,递归的过程是纵向遍历。

(1)for循环每次从startIndex开始遍历,然后用path保存取到的节点i。

(2)递归函数不断调用自己往深处遍历,总会遇到叶子节点,遇到了叶子节点就要返回。

(3)递归函数下面部分就是回溯的操作了,撤销本次处理的结果。

最终结果代码:

class Solution {// 存放单个结果path, 存放所有结果resList> res = new ArrayList<>();LinkedList path = new LinkedList<>();public List> combine(int n, int k) {combineHelper(n, k, 1);return res;}// startindex就是循环开始位置private void combineHelper(int n, int k, int startindex) {// 终止条件 if (path.size() == k){res.add(new ArrayList<>(path));return;}// 单层逻辑for (int i = startindex; i <= n ; i++ ){path.add(i);combineHelper(n, k, i + 1);path.removeLast();}}
}
  1. 剪枝优化:

(1)假设n = 4,k = 4,就四个数,还求四个数的组合,那必然只有一个组合,从2开始for循环再找其他数没有意义。所以,可以剪枝的地方就在递归中每一层的for循环所选择的起始位置,在循环中i就是循环的起始位置。也就是说for循环的开始位置到结束位置一共的元素个数

(2)过程:

  • 已经选择的元素个数:path.size();

  • 还需要的元素个数为: k - path.size();

  • 在集合n中至多要从该起始位置 : n - (k - path.size()) + 1,开始遍历。也就是说 n - (k - path.size()) + 1是最晚的起始位置,如果超过了这个位置找元素,path的元素个数不可能到达k个。这里面+1是闭区间的意思。

最终优化后的代码:

List> res = new ArrayList<>();
LinkedList path = new LinkedList<>();
public List> combine(int n, int k) {combineHelper(n, k, 1);return res;
}private void combineHelper(int n, int k, int startindex) {if (path.size() == k){res.add(new ArrayList<>(path));return;}for (int i = startindex; i <= n - (k - path.size()) + 1; i++ ){path.add(i);combineHelper(n, k, i + 1);path.removeLast();}
}

2.组合总和III

题目链接

  1. 过程图:和上一题组合类似,仍然是先取某个值,然后再从其他数中k-1个进行组合。

  1. 思路:回溯三部曲。

  • 确定递归函数参数:题目中的n和k,sum(已经收集的元素的总和也就是path里元素的总和),startIndex为下一层for循环搜索的起始位置。

  • 确定终止条件:path.size() 和 k相等且sum=n

  • 单层搜索过程: path收集每次选取的元素,sum来统计path里元素的总和。别忘了回溯。

3.代码:

class Solution {List> result = new ArrayList<>();LinkedList path = new LinkedList<>();public List> combinationSum3(int k, int n) {backTracking(n, k, 1, 0);return result;}// targetSum就是n, sum是和private void backTracking(int targetSum, int k, int startIndex, int sum) {// 减枝if (sum > targetSum) {return;}if (path.size() == k) {if (sum == targetSum) result.add(new ArrayList<>(path));return;}// 减枝 9 - (k - path.size()) + 1for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {path.add(i);sum += i;backTracking(targetSum, k, i + 1, sum);//回溯path.removeLast();//回溯sum -= i;}}
}

相关内容

热门资讯

苗族的传统节日 贵州苗族节日有... 【岜沙苗族芦笙节】岜沙,苗语叫“分送”,距从江县城7.5公里,是世界上最崇拜树木并以树为神的枪手部落...
北京的名胜古迹 北京最著名的景... 北京从元代开始,逐渐走上帝国首都的道路,先是成为大辽朝五大首都之一的南京城,随着金灭辽,金代从海陵王...
世界上最漂亮的人 世界上最漂亮... 此前在某网上,选出了全球265万颜值姣好的女性。从这些数量庞大的女性群体中,人们投票选出了心目中最美...
长白山自助游攻略 吉林长白山游... 昨天介绍了西坡的景点详细请看链接:一个人的旅行,据说能看到长白山天池全凭运气,您的运气如何?今日介绍...
猫咪吃了塑料袋怎么办 猫咪误食... 你知道吗?塑料袋放久了会长猫哦!要说猫咪对塑料袋的喜爱程度完完全全可以媲美纸箱家里只要一有塑料袋的响...