1218. 最长定差子序列(动态规划)

题意:从给定数组中提取出最长的等差序列(不一定连续),返回它的长度

暴力超时:

class Solution {
    public int longestSubsequence(int[] arr, int difference) {
        int res=1;
        for(int i=0;i<arr.length;i++) {
            int rei = 1; //初始长度
            int temp = arr[i] + difference; //下一个目标值
            for (int j = i + 1; j < arr.length; j++) {
                if (arr[j] == temp) { //找到目标值,更新长度和下一个目标值
                    rei++;
                    temp += difference;
                }
            }
            res = Math.max(rei, res); //更新最大长度
        }
        return res;
    }
}

动态规划: 每次记录以元素arr[i]为尾的等差序列的长度,在遍历过程中比较记录最大长度

class Solution {
    public int longestSubsequence(int[] arr, int difference) {
       int res=1;
       int[] dp=new int[arr.length];
       Map<Integer,Integer> map=new HashMap<>();
       for(int val:arr){
           int reV=map.getOrDefault(val-difference,0)+1;//不存在返回0
           map.put(val,reV);
           res=Math.max(res,reV);
       }
       return res;
    }
}