USACO2020DEC第二题（Daisy Chains）题解

题目
题目描述
每天，作为她绕农场行走的一部分，奶牛 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
解题思路
直接按照题目要求模拟，然后判断，累加答案。
参考代码

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n, ans = 0; cin >> n;
    int a[n + 1], sum[n + 1]; sum[0] = 0;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i]; sum[i] = sum[i - 1] + a[i];
    }
    for (int i = 1; i <= n; ++i) {
        for (int j = i; j <= n; ++j) {
            bool flag = false;
            for (int k = i; k <= j; ++k) {
                if (sum[j] - sum[i - 1] == (j - i + 1) * a[k]) {
                    flag = true; break;
                }
            }
            ans += flag;
        }
    }
    cout << ans << '\n';
    return 0;
}