LeetCode 413. 等差数列划分
413. 等差数列划分
解题思路:
双指针,指针i
指向等差数列的左端点,指针j
向前移动,查找满足等差数列的条件的数组,如果一个数列是等差数列说明每个数字之间的差值都相同。首先计算出数列前两项的差值nums[i + 1] - nums[i]
,在指针j
向后移动的过程中相邻两项的差是否相同,nums[j + 1] - nums[j]
。
当移动到等差数列的最右断点时计算等差数列的长度len = j - i + 1
,一个长度为len
的等差数列,长度不小于3的子数列有(len - 2)(len - 1) / 2
个(通过找规律的方式可以发现)。
class Solution {
public:
int numberOfArithmeticSlices(vector<int>& nums) {
int n = nums.size();
if (n < 3) return 0;
int i = 0, j = 0, ans = 0;
while(i < n - 2) {
j = i + 1;
int sub = nums[j] - nums[i];
while(j + 1 < n && nums[j + 1] - nums[j] == sub) j++;
int len = j - i + 1;
ans += ((len - 2) * (len - 1) / 2);
i = j;
}
return ans;
}
};