数据结构刷题(十九):77组合、216组合总和III
创始人
2024-05-30 00:46:19
0

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;}}
}

相关内容

热门资讯

监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
【PdgCntEditor】解... 一、问题背景 大部分的图书对应的PDF,目录中的页码并非PDF中直接索引的页码...
修复 爱普生 EPSON L4... L4151 L4153 L4156 L4158 L4163 L4165 L4166 L4168 L4...
Fluent中创建监测点 1 概述某些仿真问题,需要创建监测点,用于获取空间定点的数据࿰...
educoder数据结构与算法...                                                   ...
MySQL下载和安装(Wind... 前言:刚换了一台电脑,里面所有东西都需要重新配置,习惯了所...
MFC文件操作  MFC提供了一个文件操作的基类CFile,这个类提供了一个没有缓存的二进制格式的磁盘...