菊花链

题目

3347. 菊花链 - AcWing题库

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

花 i 有 pi 朵花瓣。

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

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

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

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

输入格式

输入的第一行包含 N。

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

输出格式

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

数据范围

1≤N≤100,
1≤pi≤1000

代码

import java.util.*;


public class Main {
    static int N = 100010;
    static int n, ans = 0, l = 1000010;
    static int[] a  = new int[1010];
    static int[] p  = new int[1010];
    public static void main(String[] args) {
        Scanner inScanner = new Scanner(System.in);
        n = inScanner.nextInt();
        for(int i = 1; i <= n; i ++)
        {
            p[i] = inScanner.nextInt();
            a[i] = a[i - 1] + p[i];
        }
        inScanner.close();
        ans = n;
        for(int i = 1; i < n; i ++)
        {
            for(int j = i + 1; j <= n; j ++)
            {
                int t = (a[j] - a[i - 1]) / (j - i + 1);
                if(t * (j - i + 1) == a[j] - a[i - 1])
                    for(int k = i; k <= j; k ++)
                    {
                        if(p[k] == t)
                        {
                            ans ++;
                            break;
                        }
                    }
            }
        }
        System.out.printf("%d", ans);
    }
}