2021-4-10 1218最长定差子序列

leetcode 1218最长定差子序列

题目转载

最长定差子序列

题目描述

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

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

输入：

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

思路一

动态规划，用2e5大小的数组记录以value结尾最大的长度，对于负数加上偏移使索引都为正数。

代码

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        int n = arr.size();
        int stdd = 1e5;//偏移，解决负数问题
        int res = 0;
        vector<int> dp(2e5);//dp[value]存储以value结尾最大的长度
        for (int i = 0; i < n; i++) {
            int idx = stdd + arr[i]; 
            dp[idx] = max(dp[idx] , dp[idx-difference] + 1);
            dp[idx] = max(dp[idx],1); 
            res = max(res, dp[idx]);
        }
        return res;
    }
};

思路二

使用map记录以value结尾的最大长度，对map进行更新

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        map<int,int>m; 
        int max =0;
        for(int i = 0; i < arr.size(); i++){
            auto it = m.find(arr[i] - difference); //寻找减去差值的元素是否在arr中
            int temp = 1;
            if(it != m.end()){
                temp = it->second + 1;
            }
             m[arr[i]] =temp;  // m.insert（）无法覆盖以及插入的值 
            max = std::max(max, temp);
        }
        return max;
    }
};