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

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

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