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;
    }
};