双指针,指针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; } };