法1:动态规划
看了题解
想法:
例如数组:A = [1,2,3,4]
- 等差数列至少有3个元素,故考虑以A[2]、A[3]结尾的等差数列。
- 以A[2]结尾的等差数列:[1,2,3] (A[2] - A[1] == A[1] - A[0])
- 以A[3]结尾的等差数列:
- [2,3,4] (A[3] - A[2] == A[2] - A[1])
- 因为A[3]与A[2]、A[1]成等差数列,又A[2]与A[1]、A[0]成等差数列,故A[3]与A[2]、A[1]、A[0]成等差数列。可以发现:若A[3]与A[2]、A[1]成等差数列,则以A[2]结尾的等差数列加上A[3]还是等差数列;故以A[3]结尾的等差数列数 = 以A[2]结尾的等差数列数 + 1 (此处加上的 1 为:A[3]与A[2]、A[1],因为:1. 以A[2]结尾的等差数列不包括[A[1], A[2]];2. 只有A[3]与A[2]、A[1]成等差数列,以A[3]结尾的等差数列数才等于(以A[2]结尾的等差数列数 + 1),否则为0
- 用dp[i]保存以A[i]结尾的等差数列个数,最后的结果为dp所有元素之和
/**
* @param {number[]} A
* @return {number}
*/
var numberOfArithmeticSlices = function(A) {
var i = 0, dp = [], sum = 0;
dp[0