题目地址
https://leetcode-cn.com/problems/arithmetic-slices/
题目描述
如果一个数列 至少有三个元素 ,并且任意两个相邻元素之差相同,则称该数列为等差数列。
例如,[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。
给你一个整数数组 nums ,返回数组 nums 中所有为等差数组的 子数组 个数。
子数组 是数组中的一个连续序列。
示例 1:
输入:nums = [1,2,3,4]
输出:3
解释:nums 中有三个子等差数组:[1, 2, 3]、[2, 3, 4] 和 [1,2,3,4] 自身。
示例 2:
输入:nums = [1]
输出:0
提示:
1 <= nums.length <= 5000
-1000 <= nums[i] <= 1000
主要思路
动态规划。定义一个dp数组,dp[i]表示以nums[i]结尾的等差子数组有多少个,这样一来,最终的结果就是遍历dp数组求和。
代码实现
javascript
var numberOfArithmeticSlices = function(nums) {
let len = nums.length, dp = new Array(len).fill(0);
for (let i = 2; i < len; i++) {
if (nums[i]-nums[i-1] == nums[i-1]-nums[i-2]) dp[i] = dp[i-1] + 1;
}
let res = 0;
for (let i = 0; i < len; i++) {
res += dp[i];
}
return res;
};
优化后的代码
let len = nums.length, cur = 0, res = 0;
for (let i = 2; i < len; i++) {
2*nums[i-1] === nums[i]+nums[i-2] ? cur++ : cur = 0;
res += cur;
}
return res;