三、最长等差数列（Weekly Contest 132）

题目描述：
给定一个整数数组 A，返回 A 中最长等差子序列的长度。

回想一下，A 的子序列是列表 A[i_1], A[i_2], …, A[i_k] 其中 0 <= i_1 < i_2 < … < i_k <= A.length - 1。并且如果 B[i+1] - B[i]( 0 <= i < B.length - 1) 的值都相同，那么序列 B 是等差的。

示例 1：

输入：[3,6,9,12]
输出：4
解释：
整个数组是公差为 3 的等差数列。
示例 2：

输入：[9,4,7,2,10]
输出：3
解释：
最长的等差子序列是 [4,7,10]。
示例 3：

输入：[20,1,15,3,10,5,8]
输出：4
解释：
最长的等差子序列是 [20,15,10,5]。

提示：

2 <= A.length <= 2000
0 <= A[i] <= 10000

这道题我一开始以为O(n^3)会超时，但是没想到整到最后自己想的超时了，也没写出来，题目不难，但是很难想出来…
我的代码就是一对于每个从-max到max开始寻找，这样就很费时，因此会超时一点也不奇怪

	public int longestArithSeqLength(int[] A) {
		if(A.length == 0){
			return 0;
		}
        Map<Integer, Integer> map = new TreeMap<>(new Comparator<Integer>() {

			@Override
			public int compare(Integer o1, Integer o2) {
				// TODO 自动生成的方法存根
				return o1.compareTo(o2);
			}
		});
        int max = A[0];
        int min = A[0];
        int index = 0;
        for (int integer : A) {
			map.put(index ++, integer);
			max = integer > max ?  integer:max ;
			min = integer < min ?  integer:min ;

		}
        
        int gongcha = max - min;
        int tems = gongcha;
        System.out.println("公差为 "+gongcha);
        System.out.println(-gongcha);
		while (tems >= -gongcha) {
			for (int i = 0; i < A.length; i++) {
				
				if(size >= A.length - i - 1){
					break;
				}
				
				int temin = A[i];
				int temindex = i;
				int result1 = 0;
				boolean flag = true;
				while(map.containsValue(temin - tems)){
					for (Map.Entry<Integer, Integer> ss: map.entrySet()) {
						if(ss.getValue() == temin - tems && ss.getKey() > temindex){
							temindex = ss.getKey();
							result1 ++;
							flag = false;
							break;
						}
					}
					if(flag){
						break;
					}else {
						temin -= tems;
						flag = true;
					}
				}
				size = size > result1 ? size :result1;
				
			}
			tems --;
		}
		return size + 1;
    }

改进后的代码，针对特定的公差进行寻找最大的size，而不是每一个都进行寻找

class Solution {
    public int longestArithSeqLength(int[] A) {
     int result = 0;
    	for (int i = 0; i < A.length; i++) {
			for (int j = i + 1; j < A.length; j++) {
				int size = 2;
				int diff = A[j] - A[i];
				int tems = A[j];
				for (int k = j + 1; k < A.length; k++) {
					if(A[k] == tems + diff){
						size ++;
						tems = A[k];
                      
					}
				}
				result = size > result ? size : result;
			}
		}
    	return result;
    }
}

可以参考一下动态规划解法

class Solution {
    public int longestArithSeqLength(int[] A) {
        int n = A.length;
        // 特判
        if(n == 0) return 0;
        // 定义哈希表，第一个键表示数组下标索引，其嵌套的哈希表用于存储该元素以不同的公差所包含的最长序列
        Map<Integer, Map<Integer, Integer>> map = new HashMap<>();
        int res = 1;
        // 遍历数组
        for(int i = 0; i < n; i++) {
            map.put(i, new HashMap<>());
            // 向前遍历，寻找不同公差的最长序列
            for(int j = i - 1; j >= 0; j--) {
                // 如果遇到了重复的元素，可以直接跳过，以为肯定不会比后面的元素能组成更长的序列
                if(map.get(i).containsKey(A[i] - A[j])) continue;
                // 获取以这两个元素差为公差的最长子序列
                int cur =  map.get(j).getOrDefault(A[i] - A[j], 0);
                // 比较答案
                res = Math.max(res, cur + 2);
                // 存入当前元素，某公差下的最长序列
                map.get(i).put(A[i] - A[j], cur + 1);
            }
        }

        return res;
    }
}