小E的怪物挑战 | 豆包MarsCode AI刷题

144 阅读3分钟

问题描述

小E在一个游戏中遭遇了一个按顺序出现的怪物挑战,每个怪物都有特定的血量 hi 和攻击力 ai。小红的初始血量为 H,攻击力为 A。规则如下:

  1. 小红可以击败一个血量和攻击力均低于她当前值的怪物;
  2. 每击败一个怪物,小红的血量与攻击力均会加上该怪物的血量和攻击力;
  3. 小红必须按照怪物的出现顺序挑战,即每次击败的怪物必须满足其血量和攻击力都高于前一个被击败的怪物。

目标:小红想知道,最多能击败多少怪物。

测试用例

示例 1: 输入:

n=3,H=4,A=5,h=[1,2,3],a=[3,2,1]n = 3, H = 4, A = 5, h = [1, 2, 3], a = [3, 2, 1]n=3,H=4,A=5,h=[1,2,3],a=[3,2,1]

输出:

111

示例 2: 输入:

n=5,H=10,A=10,h=[6,9,12,4,7],a=[8,9,10,2,5]n = 5, H = 10, A = 10, h = [6, 9, 12, 4, 7], a = [8, 9, 10, 2, 5]n=5,H=10,A=10,h=[6,9,12,4,7],a=[8,9,10,2,5]

输出:

222

示例 3: 输入:

n=4,H=20,A=25,h=[10,15,18,22],a=[12,18,20,26]n = 4, H = 20, A = 25, h = [10, 15, 18, 22], a = [12, 18, 20, 26]n=4,H=20,A=25,h=[10,15,18,22],a=[12,18,20,26]

输出:

333


解题思路

这是一个典型的动态规划(Dynamic Programming)问题,核心在于:

  1. 对每个怪物,判断是否可以击败;
  2. 记录小红击败怪物的最大可能数量;
  3. 确保击败怪物的顺序满足递增约束。

1. 数据预处理

将每个怪物的血量和攻击力打包成数组 monsters,并进行排序,确保血量和攻击力递增。

2. 状态定义

使用动态规划数组 dp,其中 dp[i] 表示以怪物 iii 为终点的最多击败怪物数量。

3. 状态转移

对于每个怪物 iii,遍历所有可能的前置怪物 jjj,如果满足以下条件:

  1. 怪物 jjj 的血量和攻击力小于怪物 iii;
  2. 小红在击败怪物 jjj 后可以击败怪物 iii。

那么:

dp[i]=max⁡(dp[i],dp[j]+1)dp[i] = max(dp[i], dp[j] + 1)dp[i]=max(dp[i],dp[j]+1)

4. 初始条件与结果

初始条件:dp[i] = 1(每个怪物都可以单独击败)
结果:max(dp) 即为小红可以击败的最多怪物数量。


代码实现

以下是基于上述思路的 Python 代码:


def solution(n: int, H: int, A: int, h: list, a: list) -> int:
    # 初始化怪物信息
    monsters = [(h[i], a[i]) for i in range(n)]
    monsters = [(0, 0)] + sorted(monsters)  # 加入虚拟初始点
    n = len(monsters)

    # 动态规划数组
    dp = [0] * n

    for i in range(1, n):
        dp[i] = 1  # 每个怪物单独击败时的初始值
        for j in range(1, i):
            if monsters[j][0] < monsters[i][0] and monsters[j][1] < monsters[i][1]:
                dp[i] = max(dp[i], dp[j] + 1)

    return max(dp)


关于 AI 使用

  1. 代码优化与完善

    • AI 初步生成的代码中,未包含对排序和动态规划的详细处理,需手动补充;
    • 对边界条件(如怪物数量为 0)的处理不够全面。
  2. 思路引导

    • AI 提供了明确的动态规划方向(如状态定义、转移方程),帮助快速锁定问题核心。
  3. 模型局限性

    • AI 在复杂排序与条件判断时,缺乏对特殊场景的综合考虑,需要通过人工优化完成细节补充。