跳到主要内容

1218.最长定差子序列

· 阅读需 2 分钟

1、题干

给你一个整数数组 arr 和一个整数 difference,请你找出并返回 arr 中最长等差子序列的长度,该子序列中相邻元素之间的差等于 difference

子序列 是指在不改变其余元素顺序的情况下,通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。

 

示例 1:

输入:arr = [1,2,3,4], difference = 1
输出:4
解释:最长的等差子序列是 [1,2,3,4]。

示例 2:

输入:arr = [1,3,5,7], difference = 1
输出:1
解释:最长的等差子序列是任意单个元素。

示例 3:

输入:arr = [1,5,7,8,5,3,4,2,1], difference = -2
输出:4
解释:最长的等差子序列是 [7,5,3,1]。

 

提示:

  • 1 <= arr.length <= 105
  • -104 <= arr[i], difference <= 104

头回见到用哈希表做dp table,记录下。

代码

/**
* @param {number[]} arr
* @param {number} d
* @return {number}
*/
var longestSubsequence = function (arr, d) {
let max = 1;
const dp = {};
for (let n of arr) {
dp[n] = (dp[n - d] || 0) + 1;
max = Math.max(max,dp[n]);
}

return max;
};