python-leetcode-解决智力问题

2140. 解决智力问题 - 力扣(LeetCode)

这道题是一个典型的 动态规划(Dynamic Programming, DP) 问题,可以使用 自底向上 的方式解决。

思路

  1. 定义状态

    dp[i] 表示从第 i 题开始,能获得的最高分数。

  2. 状态转移方程

    • 选择解决第 i
      • 这样可以获得 questions[i][0] 分,并且需要跳过 questions[i][1] 题。
      • 下一次可以从 i + questions[i][1] + 1 题开始,即 dp[i] = questions[i][0] + dp[i + questions[i][1] + 1]
    • 选择跳过第 i
      • 这样可以从 i+1 题开始,即 dp[i] = dp[i+1]
    • 取两者的最大值: dp[i]=max⁡(questions[i][0]+dp[i+questions[i][1]+1],dp[i+1])
  3. 边界条件

    • dp[n] = 0 (当超过最后一题时,得分为 0)。
  4. 计算顺序

    • 我们需要从 后往前 计算 dp[i],因为 dp[i] 依赖于 dp[i+1]dp[i + questions[i][1] + 1]

代码实现

python 复制代码
from typing import List

def mostPoints(questions: List[List[int]]) -> int:
    n = len(questions)
    dp = [0] * (n + 1)  # dp[i] 表示从第 i 题开始能获得的最高分

    for i in range(n - 1, -1, -1):  # 逆序遍历
        points, brainpower = questions[i]
        next_index = i + brainpower + 1  # 下一道可以解的题目
        dp[i] = max(points + (dp[next_index] if next_index < n else 0), dp[i + 1])

    return dp[0]

复杂度分析

  • 时间复杂度 :O(n),我们只需遍历 questions 一次,每次 O(1) 计算 dp[i]
  • 空间复杂度 :O(n),用于存储 dp 数组。

示例

输入
复制代码
questions = [[3, 2], [4, 3], [4, 4], [2, 5]]
print(mostPoints(questions))
输出
复制代码
5

优化(O(1) 空间)

我们可以只用一个变量来存储 dp[i+1],这样 dp 数组就不需要额外存储所有状态:

python 复制代码
def mostPoints(questions: List[List[int]]) -> int:
    n = len(questions)
    next_max = 0  # 相当于 dp[i+1]
    
    for i in range(n - 1, -1, -1):
        points, brainpower = questions[i]
        next_index = i + brainpower + 1
        current = max(points + (dp[next_index] if next_index < n else 0), next_max)
        next_max = current  # 更新 dp[i]
    
    return next_max

这样,我们将 空间复杂度优化为 O(1)

相关推荐
小蒋的技术栈记录1 小时前
网易大模型算法岗面经80道
算法
Ress Matthew1 小时前
重塑优化建模与算法设计:2025年大模型(LLM)在优化领域的应用盘点 - 2
算法
找不到、了2 小时前
Java排序算法之<插入排序>
java·算法·排序算法
@蓝莓果粒茶2 小时前
LeetCode第350题_两个数组的交集II
c++·python·学习·算法·leetcode·职场和发展·c#
设计师小聂!2 小时前
力扣热题100----------53最大子数组和
java·数据结构·算法·leetcode
CoovallyAIHub2 小时前
避开算力坑!无人机桥梁检测场景下YOLO模型选型指南
深度学习·算法·计算机视觉
YouQian7722 小时前
问题 C: 字符串匹配
c语言·数据结构·算法
yanxing.D2 小时前
408——数据结构(第二章 线性表)
数据结构·算法
艾莉丝努力练剑3 小时前
【LeetCode&数据结构】二叉树的应用(二)——二叉树的前序遍历问题、二叉树的中序遍历问题、二叉树的后序遍历问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
YuTaoShao3 小时前
【LeetCode 热题 100】51. N 皇后——回溯
java·算法·leetcode·职场和发展