题目描述:
给你一个整数数组 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]。
解题思路:
- 新建一个map存放每一个位置的数字对应的子序列长度
- 获取当前数字减去公差的子序列长度 val
- 如果val为空则证明当前位置之前没有跟他匹配的公差子序列,则当前位置的公差子序列长度为1
- 如果val不为空,则val加一,再进行记录
- 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 ;
}