玛朵莉与宇宙 暴力枚举+前缀和

思路：看题目数据范围，每个ai都是小于10的，那么直接暴力枚举前缀和的话是1e6的，不会超时，因此维护一个前缀和数组，对每个前缀和，枚举小于他的完全平方的个数，再用一个数组维护前缀和的完全平方的个数的值就可以了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;

const int N=1e6+10;
int a[N],m[N];

int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i],a[i]+=a[i-1];
    long long cnt=0;
    m[0]=1;
    for(int i=1;i<=n;i++){
        for(int j=0;j*j<=a[i];j++){
            cnt+=m[a[i]-j*j];
        }
        m[a[i]]++;
    }
    cout<<cnt<<endl;
    return 0;
}