描述
每天,作为她绕农场行走的一部分,奶牛 Bessie 会经过她最喜爱的草地,其中种有 N 朵花(五颜六色的雏菊),编号为 1…N(1≤N≤100),排列成一行。花 i 有 pi 朵花瓣(1≤pi≤1000)。
作为一名崭露头角的摄影家,Bessie 决定给这些花拍些照片。具体地说,对于每一对满足 1≤i≤j≤N 的花 (i,j),Bessie 会给从花 i 到花 j 之间的所有花(包括 i 和 j)拍一张照。
后来 Bessie 查看这些照片时注意到有些照片里存在「平均」的花——一朵恰好有 P 朵花瓣的花,其中 P 等于照片中所有花的花瓣数量的平均值。
Bessie 的照片中有几张存在平均的花?
输入
输入的第一行包含 N。第二行包含 N 个空格分隔的整数 p1…pN。
输出
输出存在平均的花的照片数量。
样例输入
4
1 1 2 3
样例输出
6
提示
每张仅包含一朵花的照片均会被计入答案(在这个样例中有 4 张)。另外,在这个样例中 (i,j) 为 (1,2) 和 (2,4) 所对应的照片也存在平均的花。
分析
枚举即可
代码
#include<iostream>
#include<unordered_set>
using namespace std;
typedef long long LL;
const int N = 110;
int n;
int q[N], s[N];
// 求以 i 为起点的区间有几个满足题目要求的
LL solve(int u){
LL res = 0;
unordered_set<int> hash;
for(int i = u; i <= n; i++){
hash.insert(q[i]);
if((s[i] - s[u - 1]) % (i - u + 1)) continue;
int avg = (s[i] - s[u - 1]) / (i - u + 1);
if(hash.count(avg)) res++;
}
return res;
}
int main(){
scanf("%d", &n);
for(int i = 1; i <= n; i++){
scanf("%d", &q[i]);
s[i] = s[i - 1] + q[i];
}
LL res = 0;
for(int i = 1; i <= n; i++) res += solve(i);
printf("%d\n", res);
return 0;
}
给个赞和关注吧