• 回溯算法的奥秘(LeetCode),组合77题、216题、40题


    疫情如潮水,顺势而来、又逆势而去。反反复复,希望疫情快点消失——

    回溯是递归的伴生产物,在前面的二叉树遍历中,我们发现,后续遍历就有一个回溯的思想。现在我们来仔细的学习这个算法思想。

    回溯算法结合着递归使用,主要确定三个点:

    1. 返回值,绝大多数情况是void。
    2. 结束条件,在题目中其实它就已经高数你了结束条件,隐式的。
    3. 参数,这个一下子式确定不了的,边写边填。

      首先77题:指定区间,指定每个子集的个数。

    定义一个集合用来存储子集。再定义一个子集集合存储元素。首先结束条件当让是子集长度等于k了。然后加入到ans里面。for循环用来确定递归的宽度。

    1. class Solution {
    2. List> ans = new ArrayList<>();
    3. List list = new ArrayList<>();
    4. public List> combine(int n, int k) {
    5. dfs(k, n, 1);
    6. return ans;
    7. }
    8. private void dfs(int k, int n, int p) {
    9. if (list.size() == k) {
    10. ans.add(new ArrayList<>(list));
    11. return;
    12. }
    13. for (int i = p; i <= n; i++) {
    14. list.add(i);
    15. dfs(k, n, i + 1);
    16. list.remove(list.size() - 1);
    17. }
    18. }
    19. }

    递归函数后面接上回溯,也就是下一次循环初始化。

    216题多了一个条件,子集元素相加等于指定tar.

    所以在结束的时候多一个判断,回溯的时候多一个回溯:

    1. class Solution {
    2. List> ans = new ArrayList<>();
    3. List list = new ArrayList<>();
    4. int sum = 0;
    5. public List> combinationSum3(int k, int n) {
    6. dfs(k, n, 1);
    7. return ans;
    8. }
    9. private void dfs(int k, int tar, int p) {
    10. if (list.size() == k) {
    11. if (sum == tar) {
    12. ans.add(new ArrayList<>(list));
    13. return;
    14. }
    15. }
    16. for (int i = p; i <=9; i++) {
    17. if (sum+i> tar || list.size() > k) {
    18. break;
    19. }
    20. sum += i;
    21. list.add(i);
    22. dfs(k, tar, i + 1);
    23. sum -= i;
    24. list.remove(list.size() - 1);
    25. }
    26. }
    27. }

    当子集长度大于k的时候,不满住题目

    当子集sum大于tar的时候,也不满足

    所以这里可以剪枝一下哦~

    40题就难度提升了很多,题目给定数组,并且元素不能重复使用,也就是说我们还得在递归中判断当前的元素前面的父类递归有没有使用过,这就吧难度提升了很多。

    这题的思路是这样的:

    1. 首先我们得解决重复使用这个大问题,元素是否重复使用,肯定是两个相同的元素做判断,所以我们得排序一下,方便判断是否元素相同也就是arr[i]==arr[i-1]
    2. 其次就是既然元素相同的,如何知道它是这个数组中本来就有的元素呢,
    3. 如:【1,1,1,2】,所以这里使用一个数组,专门记录当前元素的下标是否被使用
    1. List> ans = new ArrayList<>();
    2. public List> combinationSum2(int[] candidates, int target) {
    3. chak = new boolean[candidates.length];
    4. Arrays.sort(candidates);
    5. dfs(target, candidates.length, candidates, 0);
    6. return ans;
    7. }
    8. List list = new ArrayList<>();
    9. int sum = 0;
    10. boolean[] chak;
    11. private void dfs(int k, int n, int[] arr, int p) {
    12. if (sum == k) {
    13. ans.add(new ArrayList<>(list));
    14. return;
    15. }
    16. for (int i = p; i < n; i++) {
    17. if (sum + arr[i] > k) {
    18. break;
    19. }
    20. if (i > 0 && arr[i] == arr[i - 1] && !chak[i - 1]) {
    21. continue;
    22. }
    23. chak[i] = true;
    24. sum += arr[i];
    25. list.add(arr[i]);
    26. dfs(k, n, arr, i + 1);
    27. chak[i] = false;
    28. sum -= arr[i];
    29. list.remove(list.size() - 1);
    30. }
    31. }

     此时我们的回溯就有三个了:子集集合、子集的和、还有就是元素是否被使用的数组

    !chak[i - 1]判断上一个是否使用,求false的原因是,已经被回溯重置了。

  • 相关阅读:
    LeetCode50天刷题计划(Day 9—— 整数转罗马数字(20.40-22.10)
    LeetCode 80. 删除有序数组中的重复项 II
    【数据库系统概论】第四章数据库安全性
    如何查看MySQL的安装位置
    ECU的软硬件架构
    网络规划设计师之OSI七层模型之应用层
    微服务篇-C 深入理解第一代微服务(SpringCloud)_II 深入理解Eureka服务治理
    螺杆支撑座的这些特点,你知道吗?
    37 深度学习(一):查看自己显卡的指令|张量|验证集|分类问题|回归问题
    QT:QSS自定义 QAbstractScrollArea实例
  • 原文地址:https://blog.csdn.net/qx020814/article/details/128205070