1218. 最长定差子序列

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

思路: 字典

  主要目的是寻找序列中的等差数列,那么当前等差数列能否增长的因素是[能否和已经记录的序列接上].

  如diff = 1, arr[i] = 5  那么主要看i之前是否已经找到了4,如果有4,那么序列就能增长

  用字典记录已经出现的数字,并且记录数字对应的等差数列长度.

  

1 class Solution:
2     def longestSubsequence(self, arr, difference: int) -> int:
3         res_map = {}
4         for i in arr:
5             if i - difference in res_map:
6                 res_map[i] = res_map[i - difference] + 1
7             else:
8                 res_map[i] = 1
9         return max(res_map.values())

 

posted @ 2021-11-05 09:46  散装英语king  阅读(32)  评论(0编辑  收藏  举报