力扣 401 至 500 中等_力扣写多少道算法算中等水平-CSDN博客

本文链接：https://blog.csdn.net/qq_26697045/article/details/121299405

本文介绍了LeetCode413题的解决方案，主要讨论如何找到给定数组中的连续等差子序列。文章通过动态规划的方法，详细解释了如何计算连续等差数列的数量，并提供了C++代码实现。内容涵盖了等差数列的性质和如何进行有效划分。

摘要由CSDN通过智能技术生成

欢迎访问我的博客首页。

力扣 401 至 500 中等

1. LeetCode 413.等差数列划分

1. LeetCode 413.等差数列划分

相关题目《LeetCode 446.等差数列划分Ⅱ》，等差数列不必连续。
本题要找连续的等差数列，所以很简单。只需求连续的子等差数列。如果得到一个连续的长度为 n（n>=3）的等差数列，那么就可以从它中拆分出 n-2 个长度为 3 的连续的等差数列、或者划分出 n-3 个长度为 4 的连续的等差数列、…、或者划分出 1 个长度为 n 的连续的等差数列。所以共可以划分出 $\times (n-2) /2$ 个连续的等差数列。

class Solution {
public:
	int numberOfArithmeticSlices(vector<int>& nums) {
		if (nums.size() < 3)
			return 0;

		int res = 0, diff = nums[1] - nums[0], n = 2;
		for (int i = 2; i < nums.size(); i++) {
			if (nums[i] - nums[i - 1] == diff) {
				n++;
				// 1.遍历到结尾了。
				if (i == nums.size() - 1 && n >= 3)
					res += (n - 1) * (n - 2) / 2;
			}
			// 2.没有遍历到结尾，但差变了。
			else {
				if (n >= 3)
					res += (n - 1) * (n - 2) / 2;
				diff = nums[i] - nums[i - 1];
				n = 2;
			}
		}
		return res;
	}
};