跳转至

LeetCode: 1218. 最长定差子序列

1、题目描述

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

示例 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 <= 10^5
  • -10^4 <= arr[i], difference <= 10^4

2、解题思路

  • 从后面向前查找,更新当前值最大的长度即可
from collections import defaultdict


class Solution:
    def longestSubsequence(self, arr: List[int], difference: int) -> int:
        if not arr:
            return 0
        mapping = defaultdict(int)
        ans = 1
        for num in arr:
            if num - difference in mapping:
                mapping[num] = max(mapping[num - difference] + 1, mapping[num])

            else:
                mapping[num] = 1
            ans = max(ans, mapping[num])

        return ans