力扣446-等差数列划分子序列

        给你一个整数数组 nums ，返回 nums 中所有等差子序列的数目。如果一个序列中至少有三个元素 ，并且任意两个相邻元素之差相同，则称该序列为等差序列。
        例如，[1, 3, 5, 7, 9]、[7, 7, 7, 7] 和 [3, -1, -5, -9] 都是等差序列。[1, 1, 2, 5, 7] 不是等差序列。数组中的子序列是从数组中删除一些元素（也可能不删除）得到的一个序列。如[2,5,10] 是 [1,2,1,2,4,1,5,10] 的一个子序列。

       输入：int[] nums = {0,2000000000,-294967296}，输出为0；输入nums = [2,4,6,8,10]，输出为7；输入nums = [7,7,7,7,7]，输出为16；

package likou;

import java.util.HashMap;
import java.util.Map;

public class Demo446 {
	//[2,4,6,8,10]
	public int numberOfArithmeticSlices(int[] nums) {
		int length = nums.length;
		int all = 0;
		int count = 0;
		/* 1.定义一个长度为数列长度的数组
		 * 2.数组的每一个元素设为Map<Long,Integer>类型
		 */
		Map<Long,Integer>[] f = new Map[length];
		for(int i=0;i<length;i++) {
			f[i] = new HashMap<Long,Integer>();
		}
		/*
		 * 假设最后形成的等差数列为......nums[i-1],nums[i]
		 * 第一层以i作为数组的最后一个元素，再对i左边的元素进行遍历，看有哪些元素满足条件
		 * 第二层遍历以j开始，j的取值范围从0开始，知道i-1
		 * 假设输入[2,4,6,8,10]
		 * i = 1开始遍历，
		 *   i = 1时，f[i] = {2=1}
		 *   i = 2时，f[2] = {4=1,2=2} all = 1
		 *   i = 3时，f[3] = {6=1,4=1,2=3} all = 1+2=4
		 *   i = 4时，f[4] = {8=1,6=1,4=2,2=4} all = 3+1+3=7
		 */
		for(int i =0;i<length;i++){
			for(int j=0;j<i;j++) {
				long d = (long)nums[i] - (long)nums[j]; //计算数列后两个数的公差
				/*如果有值就取值，如果没有就取0
				 * 这里有一个特征就是，假设原集合已经是等差数列（最后一个元素为nums[i]），且满足条件的子序列为t个，
				 * 则该集合的后一个元素nums[i+1]加进来也满足等差数列,则符合条件的子序列数量为t+t个，即all = all+count;
				 * */
				count = f[j].getOrDefault(d, 0);//getOrDefault函数是指有值就返回，没有值就返回默认0
				all = all+count;
				int temp = count + f[i].getOrDefault(d, 0)+1;//这一步比较关键
				f[i].put(d, temp);
			}
		}
		return all;
	}
	
}