LeetCode 413.等差数列划分

LeetCode 413.等差数列划分

动态规划

if(num[i] - nums[i-1] == nums[i-1] - nums[i-2])		// 说明此三个数是等差数组

dp数组：

dp[i] : 第i个元素的标志位
最后 return accumulate(dp.begin() , dp.end() , 0);

label:标志位，对应着第i个元素连续等差数列的长度：1-3；2-4；…

​ 如果加入第i个元素后，上组等差数列仍然是等差数列，则label++;

​ 如果不是，label从0开始；

eg:

[1 , 2 , 3 , 4]

	1	2	3	4	5	0
res	0	0	1	3	6	0
label			1	1+1	2+1	0
dp : return sum				2	3	0

完整代码：

#include<numeric>
class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        if(nums.size() < 3)
            return 0;
        vector<int> dp(nums.size(),0);
        int label = 0;
        for(int i=2 ; i<nums.size() ; i++)
        {
            if(nums[i] - nums[i-1] == nums[i-1] - nums[i-2])
            {
                dp[i] = ++label;
            }
            else
            {
                label = 0;
            }
        }
        return accumulate(dp.begin() , dp.end() , 0);
    }
};