菊花链（拍照）

每天，作为她绕农场行走的一部分，奶牛 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

输入样例：
4
1 1 2 3
输出样例：
6
样例解释
每张仅包含一朵花的照片均会被计入答案（在这个样例中有 4 张）。

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

思路：暴力枚举

时间复杂度：O(n^3)

c++代码：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 110;
int n,ans;
double f[N],s[N];
bool check(int i,int j)
{
    double res=(s[j]-s[i-1])/(j-i+1);
    for (int x = i; x <= j; x ++ )
       if(f[x]==res)return true;
     return false;
}
int main()
{
    cin>>n;
    for (int i = 1; i <= n; i ++ )
    {
        cin>>f[i];
        s[i]=s[i-1]+f[i];
    }
    for (int i = 1; i <= n; i ++ )
      for(int j=i;j<=n;j++)
       if(check(i,j))ans++,cout<<i<<" "<<j<<endl;
    return cout<<ans,0;
}

Java代码：

import java.util.*;
public class Main {
    static double [] s = new double[200];
    static double a[] = new double[200];
    public static boolean check(int i,int j)
    {
        double res=(s[j]-s[i-1])/(j-i+1);
        for (int x = i; x <= j; x ++ )
        if(a[x]==res)return true;
        return false;
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        for (int i = 1; i <= N; i++) {
            a[i] = sc.nextInt();
            s[i] = s[i - 1] + a[i];
        }
        int res = 0;
        for (int i = 1; i <= N; i++) 
        {
            for (int j = i ; j <= N; j++)
            {
                if(check(i,j)==true)res++;
            }
        }
        System.out.println(res);
    }
}

思路二：用判重数组标记区间

时间复杂度：O(n^2)

c++代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 110, M = 1010;
int n;
int p[N];
bool st[M];

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ ) cin >> p[i];

    int cnt=0;
    for (int i = 1; i <= n; i ++ )
    {//枚举所有区间左端
        memset(st, 0, sizeof st);//初始化所有花的对应花瓣数是否在区间内
        int sum=0;//区间内花瓣总数
        int num=0;//区间内花数量
        for (int j = i; j <= n; j ++ )
        {//区间所有可能的右端 从小到大遍历 保证两端内所有点被记录过
            st[p[j]]=1;//区间内有该花瓣数的花
            sum+=p[j];
            num++;
            if(sum%num==0&&st[sum/num]) cnt++;//能整除 平均数花瓣的花存在区间内 计数
        }
    }
    cout << cnt;
    return 0;
}

其他大佬的解法：

思路三：哈希+枚举

时间复杂度：O(n^2)

#include<bits/stdc++.h>
using namespace std;

const int N = 105;
int a[N];
int main()
{
    int n,ans=0;
    cin>>n;
    for(int i=0;i<n;++i)  cin>>a[i];

    for(int i=0;i<n;++i){
        double sum=a[i],num=1;
        unordered_map<double,bool>mp;
        mp[a[i]]=true;
        if(mp.count(sum/num))
            ans++;
        for(int j=i+1;j<n;++j){
            sum+=a[j],num++;
            mp[a[j]]=true;
            if(mp.count(sum/num))
                ans++;
        }
    }
    cout<<ans<<endl;
    return 0;
}

Python3代码：

时间复杂度：O(n^2)

def main():
    n = int(input())
    nums = list(map(int, input().split()))

    res = 0
    for l in range(n):
        cur_sum = 0
        visited = set()
        for r in range(l, n):
            cur_sum += nums[r]
            visited.add(nums[r])
            if cur_sum % (r - l + 1) == 0:
                cur_average = cur_sum // (r - l + 1)
                if cur_average in visited:
                    res += 1
    print(res)


if __name__ == "__main__":
    main()

欢迎留言点赞

嗷嗷嗷