JavaScript|LeetCode|动态规划|413.等差数列划分

法1：动态规划
看了题解
想法：
例如数组：A = [1,2,3,4]

1. 等差数列至少有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

3. 用dp[i]保存以A[i]结尾的等差数列个数，最后的结果为dp所有元素之和

/** 
* @param {number[]} A 
* @return {number} 
*/
var numberOfArithmeticSlices = function(A) {    
    var i = 0, dp = [], sum = 0;    
    dp[0] = dp[1] = 0;    
    for(i = 2; i < A.length; i++) { // 等差数列至少是3个数        
        if(A[i] - A[i - 1] == A[i - 1] - A[i - 2]) {            
            dp[i] = dp[i - 1] + 1;        
        }        
        else {            
            dp[i] = 0;        
        }    
    }    
    for(i = 0; i < dp.length; i++) {        
        sum += dp[i];    
    }    
    return sum;
};

优化
想法：

1. 发现dp[i]的值只与dp[i - 1]有关，只需两个变量保存中间值，sum在dp[i]生成时就加上

/** 
* @param {number[]} A 
* @return {number} 
*/
var numberOfArithmeticSlices = function(A) {    
    var i = 0, pre = 0, cur = 0, sum = 0;
    // cur保存以当前元素结尾的等差数列数
    // pre保存以当前元素的前一个元素结尾的等差数列数	    
    for(i = 2; i < A.length; i++) { // 等差数列至少是3个数        
        if(A[i] - A[i - 1] == A[i - 1] - A[i - 2]) {            
            cur = pre + 1;        
        }        
        else {            
            cur = 0;        
        }        
        sum += cur;        
        pre = cur;    
    }    
    return sum;
};