等差数列划分 II - 子序列

/*
题目思路：
1. 使用一个数组dp来记录以每个元素结尾的等差子序列的数目。
2. 使用一个HashMap数组dp来记录以每个元素结尾且公差为某值的等差子序列的数目。
3. 遍历每个元素，对于每个元素再遍历其之前的所有元素，计算公差。
4. 使用该公差更新dp数组。
5. 累加所有dp数组中的值，即为所求的等差子序列的数目。
6. 使用Java Stream API来实现遍历和累加。
*/
import java.util.HashMap;
import java.util.stream.IntStream;
 
public class ArithmeticSlicesStream {
    public int numberOfArithmeticSlices(int[] nums) {
        int n = nums.length;
        if (n < 3) return 0;
        HashMap<Integer, Integer>[] dp = IntStream.range(0, n)
            .mapToObj(i -> new HashMap<Integer, Integer>())
            .toArray(HashMap[]::new);
        int[] total = {0};
        IntStream.range(0, n).forEach(i -> {
            IntStream.range(0, i).forEach(j -> {
                long diff = (long) nums[i] - (long) nums[j];
                if (diff <= Integer.MIN_VALUE || diff > Integer.MAX_VALUE) return;
                int d = (int) diff;
                int count = dp[j].getOrDefault(d, 0);
                total[0] += count;
                dp[i].put(d, dp[i].getOrDefault(d, 0) + count + 1);
            });
        });
        return total[0];
    }
}