题目描述:
给你一个整数数组 arr 和一个整数 difference,请你找出并返回 arr 中最长等差子序列的长度,该子序列中相邻元素之间的差等于 difference 。
子序列 是指在不改变其余元素顺序的情况下,通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。
提示:
1 <= arr.length <= 105
-104 <= arr[i], difference <= 104
示例:
输入:arr = [1,2,3,4], difference = 1 输出:4 解释:最长的等差子序列是 [1,2,3,4]。
解法:
先考虑一般解法。
使用数组 lon 标记,len[i]表示以当前(0 - i)序列中以 arr[i] 结尾的等差子序列的最长长度。遍历数组 arr, 遍历到 arr[i] 的时候,逆向遍历到遇到第一个等于 arr[i] - difference 的数,记为 arr[j] 此时 将lon[j] + 1 赋值到 lon[i] ,表示在此等差子序列后加入了一个元素。
但时上述方法的时间复杂度为 o(n^2) ,在有些情况下会超时,我们可以利用“以空间换时间的方式”优化性能。
考虑到 arr[i] 的取值范围在 [-1e4, 1e4] 之间,将 lon 变为长度为 20001 的数组,其中 lon[i] 的意义是当前以 ( i -10000) 结尾的 arr 等差子数组的最长长度,初始时lon各个元素都为0。例如,arr=[1, 2, 3, 4], difference = 1,遍历到arr[0]时,寻找 lon[arr[0] - difference] 的值,此时为0,表示当前以arr[0] - difference 结尾的等差子序列长度为 0,现在加入arr[i] 到这个等差子序列中,也就是赋值len[10000+arr[i]] = 1。要小心的是arr[0] -difference 可能会超过 lon 数组的范围,此时直接将 lon[10000+arr[i]]赋值为1即可。
最后返回 lon 数组的最大值即可。此方法的时间复杂度为 o(n)。
代码如下:
class Solution {
public int longestSubsequence(int[] arr, int difference) {
int[] lon = new int[20001];
int temp;
for(int i=0; i<arr.length; i++){
temp = arr[i] - difference;
if(temp >= -10000 && temp <= 10000)
lon[arr[i] + 10000] = lon[temp + 10000] + 1;
else
lon[arr[i] + 10000] = 1;
}
int max =lon[0];
for(int i=1; i<20001; i++)
if(lon[i] > max)
max = lon[i];
return max;
}
}