每日一题2021/11/05

力扣: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]。

提示:

  • \(1\) <= arr.length <= \(10^5\)
  • \(-10^4\) <= arr[i], difference <= \(10^4\)

1 <= arr.length <= 105
-104 <= arr[i], difference <= 104

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-difference
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

只需要遍历每个数字,更新对应长度即可:

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

用时和内存的情况是284ms/56.3MB,超过了7.511%/24.411%

如果单纯使用数组的话用时和内存都能进一步压缩,但是思想都是一样的。