算法-回溯-组合总和3

算法-回溯-组合总和3

1 题目概述

1.1 题目出处

https://leetcode-cn.com/problems/combination-sum-iii/submissions/

1.2 题目描述

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

所有数字都是正整数。
解集不能包含重复的组合。
示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

2 回溯法

2.1 思路

这段转自组合总和-解题思路

回溯法的解体框架是什么呢,解决一个回溯问题,实际上就是一个决策树的遍历过程。一般来说,我们需要解决三个问题:

  • 路径:也就是已经做出的选择。
  • 选择列表:也就是你当前可以做的选择。
  • 结束条件:也就是到达决策树底层,无法再做选择的条件。

其中最关键的点就是:在递归之前做选择,在递归之后撤销选择。

LinkedList result = new LinkedList();
public void backtrack(路径,选择列表){
    if(满足结束条件){
        result.add(结果);
    }
    for(选择:选择列表){
        做出选择;
        backtrack(路径,选择列表);
        撤销选择;
    }
}

所以我们这里只需要按套路来就行了,不过需要注意一些地方:

  • 要将结束条件写在for循环里,才能剪枝,否则会进行不必要递归
  • 遍历路径时发现求得的和已经和目标值n相等时,还需要确定是否满足k个数。

2.2 代码

class Solution {
    List<List<Integer>> resultList = new LinkedList<>();
    public List<List<Integer>> combinationSum3(int k, int n) {
        // 最多就是1到9数字相加为45,超过就不合法
        if(n > 45){
            return resultList;
        }
        backtrack(k, n, 1, new LinkedList<>(), 0);
        return resultList;
    }
    private void backtrack(int k, int n, int start, List<Integer> tmpList, int tmpSum){
        if(k == 0){
            // 结束条件1
            return;
        }
        LinkedList<Integer> newList = new LinkedList<>(tmpList);
        for(int i = start; i < 10; i++){
            // 选择当前数字
            int tmpSum2 = tmpSum + i;
            if(tmpSum2 > n){
                // 结束条件2,剪枝
                break;
            }

            newList.add(i);
            if(tmpSum2 == n ){
                // 结束条件3,剪枝
                if(k == 1){
                    // 只有这个时候才满足k个数之和为n
                    resultList.add(newList);
                }
                break;
            }
            
            backtrack(k-1, n, i + 1, newList, tmpSum2);

            // 不选当前数字
            newList.removeLast();
        }
    }
}

2.3 时间复杂度

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
回溯法是一种解决组合问题的有效方法。组合总和等于0的问题可以通过回溯法来解决。下面是基本的回溯法求解过程: 1. 定义一个数组用来存储符合条件的组合,以及一个变量用来记录当前组合的和。 2. 对原始数组进行排序,以便于后续的剪枝。 3. 从原始数组的第一个元素开始,依次枚举所有的元素。 4. 对于每个元素,如果当前组合的和加上该元素小于等于0,则将该元素加入到组合中,并更新当前组合的和。 5. 如果当前组合的和等于0,则将该组合加入到结果数组中。 6. 对于每个元素,如果当前组合的和加上该元素大于0,则不再继续向下搜索,直接返回上一层回溯。 7. 如果当前组合的和加上该元素等于0,则将该元素从组合中删除,并更新当前组合的和。 8. 重复步骤4-7,直到枚举完所有元素。 9. 返回结果数组。 下面是Python的代码实现: ```python def combinationSum(nums): nums.sort() # 排序 res = [] def backtrack(path, target, start): if target == 0: # 找到符合条件的组合 res.append(path) return for i in range(start, len(nums)): if nums[i] > target: break # 剪枝 if i > start and nums[i] == nums[i-1]: continue # 去重 backtrack(path+[nums[i]], target-nums[i], i+1) backtrack([], 0, 0) return res ``` 该算法的时间复杂度为O(2^n),其中n为数组的长度。由于需要遍历所有可能的组合,因此时间复杂度比较高。同时,由于需要存储所有符合条件的组合,空间复杂度也比较高。因此,在实际应用中需要考虑性能问题。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值