21天活动Zyf打卡第十天Promise9913，最长定差子序列

提交结果	执行用时	内存消耗	语言	提交时间	备注
通过	124 ms	25 MB	Python3	2021/11/05 10:10

题目描述：

给你一个整数数组 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]。

解题思路：

下文为方便叙述将 difference\textit{difference}difference 简写成 ddd。

我们从左往右遍历 arr\textit{arr}arr，并计算出以 arr[i]\textit{arr}[i]arr[i] 为结尾的最长的等差子序列的长度，取所有长度的最大值，即为答案。

令 dp[i]\textit{dp}[i]dp[i] 表示以 arr[i]\textit{arr}[i]arr[i] 为结尾的最长的等差子序列的长度，我们可以在 arr[i]\textit{arr}[i]arr[i] 左侧找到满足 arr[j]=arr[i]−d\textit{arr}[j]=\textit{arr}[i]-darr[j]=arr[i]−d 的元素，将 arr[i]\textit{arr}[i]arr[i] 加到以 arr[j]\textit{arr}[j]arr[j] 为结尾的最长的等差子序列的末尾，这样可以递推地从 dp[j]dp[j]dp[j] 计算出 dp[i]dp[i]dp[i]。由于我们是从左往右遍历 arr\textit{arr}arr 的，对于两个相同的元素，下标较大的元素对应的 dp\textit{dp}dp 值不会小于下标较小的元素对应的 dp\textit{dp}dp 值，因此下标 jjj 可以取满足 j<ij<ij<i 且 arr[j]=arr[i]−d\textit{arr}[j]=\textit{arr}[i]-darr[j]=arr[i]−d 的所有下标的最大值。故有转移方程

class Solution:
    def longestSubsequence(self, arr: List[int], difference: int) -> int:
        dp = defaultdict(int)
        for v in arr:
            dp[v] = dp[v - difference] + 1
        return max(dp.values())