leetcode 1218. 最长定差子序列

问题描述

给你一个整数数组 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]。

代码

class Solution {

public:

    int longestSubsequence(vector<int>& arr, int difference) {

        int n = arr.size(),ans = 1;

        if(n == 0)return 0;

        unordered_map<int,int> table;

        vector<int>dp(n,1);

        table[arr[0]] = 0;

        for(int i = 1; i < n; ++i)

        {

            int num = arr[i] - difference;

            if(table.count(num))//说明之前有可以连成等差数列的值

            {

                int ind = table[num];//找到该数值的坐标

                dp[i] = dp[ind] + 1;

                ans = max(dp[i],ans);

            }

            table[arr[i]] = i;

        }

        return ans;

    }

};

结果：

执行用时：304 ms, 在所有 C++ 提交中击败了76.78%的用户

内存消耗：53.3 MB, 在所有 C++ 提交中击败了47.48%的用户

巴特西

leetcode 1218. 最长定差子序列

问题描述

代码

最新文章

热门文章