acwing 3347. 菊花链 | 前缀和

题目传送门

题目描述

每天,作为她绕农场行走的一部分,奶牛 Bessie 会经过她最喜爱的草地,其中种有 NN 朵花(五颜六色的雏菊),编号为 1…N1…N,排列成一行。

花 ii 有 pipi 朵花瓣。

作为一名崭露头角的摄影家,Bessie 决定给这些花拍些照片。

具体地说,对于每一对满足 1≤i≤j≤N1≤i≤j≤N 的花 (i,j)(i,j),Bessie 会给从花 ii 到花 jj 之间的所有花(包括 ii 和 jj)拍一张照。

后来 Bessie 查看这些照片时注意到有些照片里存在「平均」的花——一朵恰好有 PP 朵花瓣的花,其中 PP 等于照片中所有花的花瓣数量的平均值。

Bessie 的照片中有几张存在平均的花?

输入格式

输入的第一行包含 NN。

第二行包含 NN 个空格分隔的整数 p1…pNp1…pN。

输出格式

输出存在平均的花的照片数量。

数据范围

1≤N≤1001≤N≤100,
1≤pi≤10001≤pi≤1000

输入样例:

4
1 1 2 3

输出样例:

6

样例解释

每张仅包含一朵花的照片均会被计入答案(在这个样例中有 4 张)。

另外,在这个样例中 (i,j)为 (1,2)和 (2,4)所对应的照片也存在平均的花。

前缀和 + 滑动窗口

分析

根据题意做就行

代码

#include<iostream>
#include<cstdio>
using namespace std;
const int N = 110;
int a[N], s[N];
int n;
int main()
{
	cin >> n;
	for(int i = 1; i <= n; i++) cin >> a[i], s[i] = s[i-1] + a[i];
	
	int cnt = 0;
	for(int len = 1; len <= n; len++)
	{
		for(int l = 1; l + len-1 <= n; l++)
		{
			int r = l + len - 1;
			int sum = s[r] - s[l-1];
            
			for(int k = l; k <= r; k++)
			{
				if(a[k] * (r-l+1) == sum)
				{
					cnt++;
					break;
				}
			}
		}
	}
	cout << cnt;
	return 0;
}

时间复杂度

参考文章

posted @ 2022-03-15 11:33  VanHope  阅读(70)  评论(0编辑  收藏  举报