菊花链（寒假每日一题 39）-CFANZ编程社区

菊花链（寒假每日一题 39）

每天，作为她绕农场行走的一部分，奶牛 Bessie 会经过她最喜爱的草地，其中种有菊花链（寒假每日一题 39）_i++ 朵花（五颜六色的雏菊），编号为菊花链（寒假每日一题 39）_i++_02 ，排列成一行。

花菊花链（寒假每日一题 39）_hash_03 有菊花链（寒假每日一题 39）_i++_04 朵花瓣。

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

具体地说，对于每一对满足菊花链（寒假每日一题 39）_枚举_05 的花菊花链（寒假每日一题 39）_数据_06 ，Bessie 会给从花菊花链（寒假每日一题 39）_hash_03 到花菊花链（寒假每日一题 39）_数据_08 之间的所有花（包括和）拍一张照。

后来 Bessie 查看这些照片时注意到有些照片里存在「平均」的花——一朵恰好有菊花链（寒假每日一题 39）_枚举_11 朵花瓣的花，其中等于照片中所有花的花瓣数量的平均值。

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

输入格式
输入的第一行包含菊花链（寒假每日一题 39）_i++ 。

第二行包含菊花链（寒假每日一题 39）_i++ 个空格分隔的整数菊花链（寒假每日一题 39）_hash_15 。

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

数据范围
菊花链（寒假每日一题 39）_数据_16

输入样例：

4
1 1 2 3

输出样例：

样例解释
每张仅包含一朵花的照片均会被计入答案（在这个样例中有菊花链（寒假每日一题 39）_#include_17 张）。

另外，在这个样例中菊花链（寒假每日一题 39）_数据_06 为菊花链（寒假每日一题 39）_hash_19 和菊花链（寒假每日一题 39）_hash_20 所对应的照片也存在平均的花。

#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;
}

0 条评论