(中等)1218、最长定差子序列-字典dp

最新推荐文章于 2024-09-14 18:52:56 发布

_张员外_

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量97

点赞数

文章标签： leetcode 算法 dict python

本文链接：https://blog.csdn.net/weixin_42124486/article/details/121161803

版权

本文介绍了一种解决LeetCode上的'最长等差子序列'问题的优化算法。通过创建一个字典存储以每个数字结尾的最长等差子序列长度，避免了暴力求解导致的时间复杂度过高。在遍历数组时，利用字典更新每个数字的最长子序列长度，最后返回字典中最大值，得到最长子序列的长度。

摘要由CSDN通过智能技术生成

题目描述

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

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

来源：力扣（LeetCode）
链接：题目描述详情与示例https://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-differencehttps://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-difference

思路

一开始，直接两重循环判断每个数形成的等差数列的长度，由于时间复杂度过高，转换思路。考虑每个数字结尾的等差数列的长度配以字典的记录方式，这样能减少大量的循环。

具体思路如下：

以字典 dp为存储方式，dp[num] 表示以数字num为结尾的最长的等差子序列的长度。

那么，对于数组中数字arr[i]有,dp[arr[i]] = dp[arr[i] - difference]

其中，python3中字典的每个键值 key => value 对用冒号 : 分割，每个对之间用逗号(,)分割，整个字典包括在花括号 {} 中，格式：

d = {key1 : value1, key2 : value2, key3 : value3 }

字典的其他语法

class Solution:
    def longestSubsequence(self, arr: List[int], difference: int) -> int:
        dicts = {}
        for i in range(len(arr)):
                if arr[i]-difference in dicts:
                    dicts[arr[i]] = dicts[arr[i]-difference] + 1
                else:
                    dicts[arr[i]] = 1
        return max(dicts.values())