力扣 401 至 500 中等

欢迎访问我的博客首页。

---

1. LeetCode 413.等差数列划分

---

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