其他分享
首页 > 其他分享> > Leetcode 413. 等差数列划分

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

标签:nums,int,ans,len,413,Leetcode,等差数列,指针
来源: https://www.cnblogs.com/gxsoar/p/16083866.html