AtCoder ABC 369 C 题题解

最新推荐文章于 2024-10-02 23:40:03 发布

King童年回忆

最新推荐文章于 2024-10-02 23:40:03 发布

阅读量300

点赞数 9

文章标签： c语言算法 c++

本文链接：https://blog.csdn.net/2301_76723927/article/details/141783235

版权

题目传送门

C - Count Arithmetic Subarrays (atcoder.jp)

题解：

本题可以先预处理好 $i$ 与 $i+1$ 之间的差值。首先每个数都是一个等差数列，接着，每两个数也是一个等差数列，然后可以看一个数字持续了几位，如果持续了 $x$ 位，则说明从 $add$ （这个数字的首位置）到 $add+x$ 以前都是等差数列，则total加上 $x*(x+1)/2$ ，即为1加到 $x$ 的和。（不理解见代码）

AC代码：

//atcoder.jp/contests/abc369/tasks/abc369_c
#include <bits/stdc++.h>
#define int long long
using namespace std;
int n;
int a[200005];
int c[200005];
int total;
signed main(){
	cin>>n;
	cin>>a[1];
	for(int i=2;i<=n;i++){
		cin>>a[i];
		c[i]=a[i]-a[i-1];
	}
	total=n;
	int length=0;
	int now=c[2];
	for(int i=2;i<=n;i++){
		if(c[i]==now){
			++length;
		}
		else{
			total+=length*(length+1)/2;
			length=1;
			now=c[i];
		}
	}
	total+=length*(length+1)/2;
	cout<<total<<endl;
	return 0;
}