1027. 最长等差数列

NOTE

本文带有 todo 标签,表示尚未完成本人复刷。代码与推导已按公开题目和参考实现整理;刷完后可删除 frontmatter 中的 todo

  • LeetCode:原题1
  • 难度:中等
  • 归类:动态规划(数组、哈希表、二分查找、动态规划)
  • 主解法:动态规划

先给结论

这道题的核心是利用有序性或答案单调性,每轮排除一半不可能成为答案的候选区间。面试时先说清楚“状态表示什么、为什么能排除其他候选、何时更新答案”,再写代码;只报出“用 二分查找”还不足以证明理解。

题目描述

题目要求完成“最长等差数列”。输入包括 nums(integer[]),需要返回 integer。完整限制以原题为准,解题时重点利用这些特征:数组、哈希表、二分查找、动态规划。

示例:

输入:nums = [3,6,9,12]
输出:4
解释:
整个数组是公差为 3 的等差数列。

1.dp 数组含义

输入是 nums(integer[])。dp[i] 表示处理到第 i 个位置或规模为 i 的子问题答案;若实现采用记忆化搜索,缓存键表达的是同一个状态。状态必须能够直接推出题目最终答案,不能只把 dp 笼统说成“缓存”。

2. 确定状态转移方程

枚举当前可做的选择,把问题转移到规模更小且已经可求的状态,再根据题意取最大值、最小值、求和或判断可行性。其因果链是:旧状态已经正确 → 当前选择补上最后一步 → 合并得到当前状态。参考实现中的分支就是这条转移方程的代码形式。

3.dp 数组如何初始化

初始化来自状态定义:空前缀、越界位置或没有作出选择时使用题目的单位值;不可能状态使用 Infinity-Infinity 或明确的哨兵。不要为了“方便”统一填 0,否则最值类问题可能把不可达状态当成合法答案。

4. 确定遍历顺序

循环开始时,答案如果存在,一定仍在当前候选区间中。表格 DP 要让被依赖的行、列先计算;记忆化搜索则由递归按需展开,并在返回前写入缓存。若做一维空间压缩,遍历方向还要保证本轮不会覆盖仍需读取的旧状态。

5. 举例打印dp 数组

示例:

输入:nums = [3,6,9,12]
输出:4
解释:
整个数组是公差为 3 的等差数列。
阶段状态含义结果
初始化空前缀、边界或起点写入基础值
转移读取已经完成的更小状态按当前选择合并
结束最终位置或全部状态得到题目答案

手算时至少核对一个基础状态和一次非平凡转移;它们与代码一致,才说明状态定义成立。

代码实现

参考实现来源:JoshCrozier/leetcode-javascript,按本文结构重新整理;原项目采用 MIT License

JavaScript 实现

/**
 * @param {number[]} nums
 * @return {number}
 */
var longestArithSeqLength = function(nums) {
  const dp = new Array(nums.length).fill().map(() => new Map());
  let result = 2;

  for (let i = 1; i < nums.length; i++) {
    for (let j = 0; j < i; j++) {
      const diff = nums[i] - nums[j];
      const prevLength = dp[j].get(diff) || 1;
      dp[i].set(diff, prevLength + 1);
      result = Math.max(result, prevLength + 1);
    }
  }

  return result;
};

代码与思路对照

阶段对应代码作用
入口longestArithSeqLength接收题目输入,入口签名与 LeetCode 元数据一致
初始化var longestArithSeqLength = function(nums)const dp = new Array(nums.length).fill().map(() => new Map());let result = 2;const diff = nums[i] - nums[j];建立后续推进所需的边界、缓存或答案变量
核心推进for (let i = 1; i < nums.length; i++)for (let j = 0; j < i; j++)落实“根据中点是否满足条件移动左、右边界;边界定义和循环条件必须配套”
输出return result只返回已经满足状态定义的最终结果

读代码时应把每个判断还原成“它排除了什么状态或完成了哪次转移”,而不是只记变量名。

复杂度分析

  • 时间复杂度:O(n × (d + n))。
  • 空间复杂度:O(n × d)。

复杂度必须按“状态数量 × 每个状态处理成本”分析;如果存在排序、堆操作、递归深度或结果数组,需要单独计入,不能只看最外层循环。

面试官递进追问

1. 为什么本题适合 二分查找?

因为利用有序性或答案单调性,每轮排除一半不可能成为答案的候选区间。 为什么问: 检查你是在识别性质,还是只凭题号背模板。回答要抓住可被利用的单调性、重复子问题或数据结构约束。

2. 代码维护的核心状态是什么?

维护仍可能包含答案的搜索区间,比较中点后只保留满足题意的一侧。 为什么问: 状态定义决定代码是否可证明。回答时要让每个变量都能对应到题意。

3. 这个实现依赖什么不变量?

循环开始时,答案如果存在,一定仍在当前候选区间中。 为什么问: 不变量比逐行复述代码更能证明理解,回答要说明它在初始化、推进和结束时都成立。

4. 为什么这样推进不会漏掉答案?

根据中点是否满足条件移动左、右边界;边界定义和循环条件必须配套,被跳过的状态已经由题目性质证明不可能更优或已经处理完成。 为什么问: 检查正确性证明,重点不是“指针这样写”,而是“为什么可以排除”。

5. 最危险的边界是什么?

最容易错在开闭区间混用、边界没有越过 mid,以及重复元素下没有继续收缩。 为什么问: 检查代码能否一次通过,回答要给出具体失败场景而不是只说“注意边界”。

6. 复杂度还能优化吗?

先看瓶颈来自状态数量、每次转移成本还是辅助结构操作;只有其中一项能被减少时,复杂度才可能继续下降。 为什么问: 检查你能否从成本组成出发优化,而不是机械背最优复杂度。

7. 如果输入规模扩大或数据改成流式,方案怎么变?

要判断当前算法是否需要随机访问全部输入;若只依赖有限历史状态,可以做滚动或在线维护,否则需要分块、外存或调整数据结构。 为什么问: 检查算法理解能否迁移到工程约束。

8. 有哪些替代解法,如何取舍?

可以从暴力枚举开始,再比较排序、哈希、搜索或动态规划等方案;取舍标准是时间、空间、是否修改输入和实现复杂度。 为什么问: 检查横向比较能力。回答要说明替代方案为什么更慢或在什么条件下更合适。

常见错误回答

  • “这题套 二分查找 模板即可”:只有结论,没有说明题目性质和排除依据。
  • 逐行朗读代码:没有建立状态、不变量和正确性因果链。
  • 只报 O(n)O(n²):没有解释每个状态被访问多少次,也容易漏掉排序或辅助结构。
  • 把示例能跑通当成证明:示例只能帮助检查,不能覆盖重复值、空输入和极端边界。

可迁移总结

  • 核心关键词:二分查找、状态定义、不变量、推进规则、边界。
  • 一句话本质:利用有序性或答案单调性,每轮排除一半不可能成为答案的候选区间。
  • 思考链路:识别题目性质 → 定义状态 → 证明推进安全 → 确定更新时机 → 检查边界与复杂度。
  • 1 分钟回答:说清题型、核心状态、推进规则和复杂度。
  • 3 分钟回答:补上正确性依据、示例和两个关键边界。
  • 10 分钟回答:写出代码,并比较替代方案、空间优化和工程限制。

刷题后自测

先只回答第 1 题,再展开后续问题:

  1. 不看代码,你能用一句话说出本题的不变量吗?
  1. 如果把示例中的重复值、空输入或边界值换掉,哪一行代码最先受到影响?
  1. 不改变正确性的前提下,你能写出一种替代方案并比较复杂度吗?