66. 魔法甜点之和:小包的新挑战 | 豆包MarsCode AI刷题

72 阅读5分钟

题目解析

题目要求通过选择若干个甜点并可选择性地将其喜爱值改为阶乘形式,找到满足总和恰好等于给定预期值 S 的不同选择方案数。具体来说,给定每个甜点的初始喜爱值,可以使用魔法棒将其变为该值的阶乘。魔法棒的使用次数有上限 M,每个甜点只能使用一次魔法棒。我们需要计算在给定的甜点集合中,有多少种方式能选择一些甜点,使得它们的喜爱值之和恰好为 S。

在解决该问题时,我们需要考虑到的主要因素包括:

  • 魔法棒的使用限制:最多使用 M 次魔法棒,且每个甜点最多使用一次魔法棒。
  • 每个甜点的喜爱值有两种状态:原始值和阶乘值。
  • 我们的目标是找到所有可能的组合,使得其总和恰好为预期的值 S。

解题思路

  1. 动态规划/回溯:这是一个典型的背包问题变种。通过递归回溯的方法,可以尝试每个甜点的两种选择(原始值或者阶乘值)来计算所有可能的组合。
  2. 递归选择:对于每个甜点,我们可以选择其原始值,也可以选择其阶乘值,递归地遍历所有的选择,直到满足条件或者终止。
  3. 剪枝:为了减少不必要的递归,我们在递归过程中进行剪枝。如果当前选择的总和已经超过目标 S,或者使用的魔法棒次数超过 M,则直接停止当前递归路径。

代码详解

以下是我写的代码实现,它通过回溯的方式计算所有可能的选择方案:

def solution(N, M, S, loves):
    count = 0

    def factorial(n):
        if n == 0:
            return 1
        result = 1
        for i in range(1, n + 1):
            result *= i
        return result

    def backtrack(index, current_sum, used_magic):
        nonlocal count

        # 如果当前和超过 S,或者使用的魔法棒超过了 M,直接返回
        if current_sum > S or used_magic > M:
            return

        # 如果当前和恰好为 S,计数加一
        if current_sum == S:
            count += 1

        # 遍历剩余的甜点
        for i in range(index, N):
            # 选择当前甜点的原始喜爱值
            backtrack(i + 1, current_sum + loves[i], used_magic)

            # 选择当前甜点的阶乘值
            factorial_value = factorial(loves[i])
            backtrack(i + 1, current_sum + factorial_value, used_magic + 1)

    # 开始回溯
    backtrack(0, 0, 0)
    return count
  1. factorial函数:计算给定数字的阶乘。该函数用于转换每个甜点的喜爱值,作为第二种选择。
  2. backtrack函数:是回溯的核心部分。它递归地遍历所有的甜点,分别选择原始值和阶乘值,并根据当前总和和魔法棒使用次数进行判断。如果总和恰好为 S,则计数加一。
  3. 剪枝条件:我们在递归中加上了剪枝条件——当当前总和已经超过目标值 S 或者使用的魔法棒次数超过 M 时,就不再继续深入该路径。

知识总结

  1. 回溯法:回溯是一种用于遍历所有可能解的算法策略。通过递归地构造解空间树,并在搜索过程中进行剪枝,可以有效避免无效的搜索。该方法适用于组合问题、排列问题等。通过本题的练习,我加深了对回溯算法的理解。对于类似的组合优化问题,回溯法是一个重要的解题思路。学习过程中,可以通过分析问题的约束条件来决定是否需要剪枝,以提高算法效率。

  2. 剪枝技巧:在回溯中,剪枝是非常重要的技巧之一,它可以大大减少递归树的大小。通过提前判断某些条件(如当前和已经超过目标,或者魔法棒使用次数已超出限制),可以提前终止不必要的递归路径,从而提高效率。

  3. 阶乘运算:阶乘的增长速度非常快,因此需要注意阶乘值可能会非常大,尤其是对于较大的数字,可能导致计算量的急剧增加。因此在实际应用中,我们要尽量避免计算过大的阶乘,或者使用模运算来处理大数问题。

  4. 重视递归与递推的关系:递归是回溯的基础,理解递归的流程和回溯过程的内存使用至关重要。每次递归都会有一个新的状态,而回溯就是通过选择不同的路径来遍历所有可能的解。

  5. 错题集与总结:通过在豆包MarsCode AI上刷题,我发现了自己在回溯法的应用上还有很多可以优化的地方,特别是剪枝的判断条件。建议在做题过程中记录错题,并结合题目解析分析原因,逐步提高算法技巧。

工具运用

豆包MarsCode AI刷题功能不仅能够提供题目的练习和解题思路,还能帮助我快速发现自己的知识盲点。例如,在这道题中,我利用它的代码解析和讨论区的解题方法,深入理解了回溯法的剪枝技巧。此外,结合书籍和其他在线资源(如Leetcode、AcWing等),可以进一步强化回溯法的应用能力,提高解题效率。