LeetCode——1218. 最长定差子序列(Java)

题目描述

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

题解思路

返回最长等差序列的长度,公差为difference,所以我们可以采用动态规划来解决

从第一个元素开始,依次减少difference然后保存再Map中,这样就可以记录等差数列的长度

正确代码

    public int longestSubsequence(int[] arr, int difference) {
        int ans = 0;
        Map<Integer, Integer> dp = new HashMap<Integer, Integer>();
        for (int v : arr) {
            dp.put(v, dp.getOrDefault(v - difference, 0) + 1);
            ans = Math.max(ans, dp.get(v));
        }
        return ans;
    }

总结

虽然利用dp思路确实容易想到,但是用每个元素去减去difference来记录还是值得学习的

如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论吗,各自努力,最高处见
posted @ 2021-11-05 09:10  21岁还不是架构师  阅读(62)  评论(0编辑  收藏  举报