算法练习:最长定差子序列

题目描述:

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

解题思路:

1. 新建一个map存放每一个位置的数字对应的子序列长度
2. 获取当前数字减去公差的子序列长度 val
3. 如果val为空则证明当前位置之前没有跟他匹配的公差子序列,则当前位置的公差子序列长度为1
4. 如果val不为空,则val加一,再进行记录
5. map中val最大值就是最大公差子序列长度

解题代码:

public  int longestSubsequence(int[] arr, int difference) {
        Map<Integer,Integer> map = new HashMap<Integer,Integer>();
        int res = 1;
        for(int i:arr){
            Integer val = map.get(i-difference);
            if(val == null){
                map.put(i,1);
            }else{
                map.put(i,++val);
                res = Math.max(res,val);
            }
        }
        return res ;
    }