思路
1e5的数据显然是不可暴力枚举的(不过你可以选择相信牛客神机)。 我们用s[r]-s[l-1]表示[l,r]的区间和,看数据范围我们知道s[i]最多为1e6, 也就是说我们要的平方和最多不超过1e3,那样的话我们就可以试着枚举n以及平方和, 然后记录值等于的前缀和的个数即可得到答案。
代码
#pragma GCC optimize("Ofast", "inline", "-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define int long long
using namespace std;
const int N=2e5+7;
const int mod=1e9+7;
//int read(){ int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-') f=f*-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}
int n,v,s[N],num[N*10],ans=0;
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
cin>>v;
s[i]=s[i-1]+v;
}
num[0]=1;
for(int i=1;i<=n;i++){
for(int j=0;j*j<=s[i];j++) ans+=num[s[i]-j*j];
num[s[i]]++;
}
cout<<ans<<"\n";
return 0;
}