leetcode-go/leetcode/hard/446.arithmetic-slices-ii-subsequence.go

func numberOfArithmeticSlices(A []int) (cnt int) {
	n := len(A)
	if n < 3 {
		return
	}
	dp := make([]map[int]int, n)
	for i := 0; i < n; i++ {
		dp[i] = make(map[int]int)
	}
	for i := 1; i < n; i++ {
		for j := 0; j < i; j++ {
			diff := A[i] - A[j]
			inc := dp[j][diff] + 1
			cnt += inc - 1
			dp[i][diff] = dp[i][diff] + inc
		}
	}
	return
}