C/C++教程

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;
    }
};
这篇关于Leetcode 413. 等差数列划分的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!