字节一面原题,之前刷过,遇到时一点印象都没有!

2023-12-28 15:17:28 浏览数 (1)

今天我们来看一下 LeetCode 36:组合总和,这道题目和对应的解法。

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

示例 1:

代码语言:javascript复制
输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2   2   3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

在深入解析之前,了解一些基础概念是很重要的:

  • 回溯算法:一种通过探索所有可能的候选解来找出所有解的方法,如果候选解被证明不是一个解(或者至少不是最后一个解),回溯算法会丢弃它,并回退到之前的步骤,尝试其他的候选解。
  • 递归:方法或函数在其定义或说明中调用自身的一种程序设计技巧。

问题描述

“组合总和”问题要求我们找出所有可以使数字总和等于给定目标数的组合,这些组合是从给定的一个无重复元素数组中选择出来的。同一个数字可以无限次地被选取。

代码解析

类和全局变量
  • Solution:类名称。
  • List<List<Integer>> res:用来存储所有可能的组合。
方法 combinationSum
  • 参数:一个整数数组 candidates 和一个目标数 target
  • 功能:初始化路径(path)并开始回溯。
  • 返回:所有可能使得和等于 target 的组合。
方法 backtrack
  • 参数:nums(候选数字数组),target(目标数),path(当前路径),start(探索的起始位置)。
  • 功能:执行回溯算法的核心逻辑。
回溯算法的核心步骤
  1. 结束条件
    • 如果 target < 0,说明当前路径不可能形成解,返回。
    • 如果 target == 0,找到一个有效组合,将其加入结果中。
  2. 选择列表
    • 使用 for 循环从 start 开始选择候选数字,确保数字不会重复选取。
  3. 做出选择
    • 将当前数字 nums[i] 添加到路径中。
  4. 递归调用
    • 递归地调用 backtrack,目标值减去当前数字 nums[i],同时 start 设置为当前索引 i
  5. 撤销选择
    • 从路径中移除最后添加的数字,回溯到之前的状态。

解题思路

  1. 初始化一个空路径和结果列表。
  2. 从第一个数字开始,逐个尝试将 candidates 中的数字加入路径。
  3. 如果加入一个数字后,路径和等于目标值,则将该路径加入结果列表。
  4. 如果路径和小于目标值,继续添加更多的数字。
  5. 如果路径和超过目标值,放弃该路径,尝试其他选择。
  6. 使用递归和回溯来遍历所有可能的路径。

示例

假设 candidates = [2, 3, 6, 7]target = 7,我们的目标是找出所有组合,使得各个数字的和为 7。可能的组合包括 [2, 2, 3][7]

总结

通过递归和回溯,我们可以有效地探索所有可能的组合,找到所有使数字总和等于目标值的组合。代码中的每个步骤都是为了维护当前路径的状态,确保所有可能的组合都被探索到。

参考代码:

代码语言:javascript复制
// 登录 AlgoMooc 官网获取更多算法图解
// https://www.algomooc.com
// 作者:程序员吴师兄
// 代码有看不懂的地方一定要私聊咨询吴师兄呀
class Solution {

    List<List<Integer>> res = new ArrayList<>();

    public List<List<Integer>> combinationSum(int[] candidates, int target) {

        List<Integer> path = new ArrayList<>();

        backtrack( candidates , target , path , 0 );

        return res;
    }

    // 1、画出递归树,找到状态变量(回溯函数的参数)
    // start 表示递归时正在访问的数组元素下标
    // nums 表示当前集合中的元素
    // target 表示想在当前区间拼凑出的目标值
    // path 表示选择的路径
    private void backtrack(int[] nums,int target,List<Integer> path,int start) {
    
        // 2、寻找结束条件,由于回溯算法是借助递归实现,所以也就是去寻找递归终止条件
        if (target < 0) {
            // 一些逻辑操作(可有可无,视情况而定)
            // 比如,在 N 皇后问题中,在这一步把数据加入到了结果里面
            return;
        }

        if(target == 0 ){
            
            // 找到一个组合了
            res.add(new ArrayList(path));

            return;
        }
            
        // 3、确定选择列表,即需要把什么数据存储到结果里面
        // for 循环就是一个选择的过程
        // i 等于 start 表示,后续可以选的元素一开始只能从 start 开始
        // 比如 nums = [2,3,6,7]
        // i = 1,指向了元素 3 ,表示当前后续选择的过程中,只能从 3 开始选,可以重复选 3 ,但无法选 2 了
        // i = 2,指向了元素 6 ,表示当前后续选择的过程中,只能从 6 开始选,可以重复选 6 ,但无法选 2、3 了
        for (int i = start ; i < nums.length ; i   ) {
            
            // 当前路径上可以把 nums[i] 加上
            path.add(nums[i]);
        
            // 一些逻辑操作(可有可无,视情况而定)
            // 4、判断是否需要剪枝,去判断此时存储的数据是否之前已经被存储过
            // 需要剪枝
            // 此时,目标值 target,已经从 target 变成了 target - nums[i]
            // 接下来需要去【某个区间中】拼凑 target - nums[i]
            // 由于 同一个 数字可以 无限制重复被选取
            // 当前正在使用 nums[i],那么为了拼凑 target - nums[i],依旧可以继续从使用 nums[i] 开始
            // 而 i 前面的元素,比如 num[i-1]、 num[i-2]无法继续使用,实现了剪枝操作
            int nowposition = i ; 

            // 5、做出选择,递归调用该函数,进入下一层继续搜索
            // 递归
            backtrack(nums, target - nums[i] , path , nowposition );
        
            // 一些逻辑操作(可有可无,视情况而定)

            // 6、撤销选择,回到上一层的状态
            path.remove(path.size()-1);
        }
    }
}

0 人点赞