Leetcode 1228.最长定差子序列
给你一个整数数组 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]。
提示:
1 <= arr.length <= 105
题解:先说一下,这个difference指的是arr[i]-arr[i-1],不能是arr[i-1]-arr[i]。 我们只需要用map存一下arr[i]这个元素的最长子序列是多少就行了。
import java.util.Map;
import java.util.TreeMap;
class Solution {
public int longestSubsequence(int[] arr, int difference) {
Map<Integer,Integer> map=new TreeMap<>();
map.put(arr[0], 1);
int ans=1;
for(int i=1;i<arr.length;i++)
{
map.put(arr[i], map.getOrDefault(arr[i]-difference, 0)+1);
ans=Math.max(ans, map.get(arr[i]));
}
return ans;
}
}