JAVA程序设计： 等差数列划分 II - 子序列（LeetCode：446）

如果一个数列至少有三个元素，并且任意两个相邻元素之差相同，则称该数列为等差数列。

例如，以下数列为等差数列:

1, 3, 5, 7, 9
7, 7, 7, 7
3, -1, -5, -9
以下数列不是等差数列。

1, 1, 2, 5, 7
 

数组 A 包含 N 个数，且索引从 0 开始。该数组子序列将划分为整数序列 (P0, P1, ..., Pk)，P 与 Q 是整数且满足 0 ≤ P0 < P1 < ... < Pk < N。

 

如果序列 A[P0]，A[P1]，...，A[Pk-1]，A[Pk] 是等差的，那么数组 A 的子序列 (P0，P1，…，PK) 称为等差序列。值得注意的是，这意味着 k ≥ 2。

函数要返回数组 A 中所有等差子序列的个数。

输入包含 N 个整数。每个整数都在 -231 和 231-1 之间，另外 0 ≤ N ≤ 1000。保证输出小于 231-1。

 

示例：

 

输入：[2, 4, 6, 8, 10]

输出：7

解释：
所有的等差子序列为：
[2,4,6]
[4,6,8]
[6,8,10]
[2,4,6,8]
[4,6,8,10]
[2,4,6,8,10]
[2,6,10]
 

思路：首先这道题的n是1000，因此你的暴力枚举是行不通的，一份官方题解（感觉写的比我要讲得好）：官方题解

class Solution {
	public int numberOfArithmeticSlices(int[] A) {
        int n=A.length;
        long ans=0;
        Map<Integer,Integer>[] cnt=new Map[n];
        for(int i=0;i<n;i++) {
        	cnt[i]=new HashMap<>();
        	for(int j=0;j<i;j++) {
        		long delta=(long)A[i]-(long)A[j];
        		if(delta<Integer.MIN_VALUE || delta>Integer.MAX_VALUE)
        			continue;
        		int diff=(int)delta;
        		int sum1=cnt[j].getOrDefault(diff, 0);
        		int sum2=cnt[i].getOrDefault(diff, 0);
        		cnt[i].put(diff, sum1+sum2+1);
        		ans+=sum1;
        	}
        }
        return (int)ans;
    }
}