Leecode 1218. 最长定差子序列——Leecode每日一题系列

Posted 2021-12-08 来老铁干了这碗代码

今天是坚持每日一题打卡的第十天

---

题目链接：https://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-difference/

---

题解汇总：https://zhanglong.blog.csdn.net/article/details/121071779

---

题目描述

给你一个整数数组 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，状态转移方程为：$dp[i]=dp[i-difference]+1$

用unordered_map做存储，这样不用考虑i为负的情况。

---

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        unordered_map<int, int>dp;
        int MAX = -1;
        for (auto i : arr) {
            dp[i] = dp[i - difference] + 1;
            MAX = max(MAX, dp[i]);
        }
        return MAX;
    }
};

---

   ——锻炼发现事物统一性和功能性的能力，这样会使代码简洁、干净并且功能强大。