思路

1e5的数据显然是不可暴力枚举的(不过你可以选择相信牛客神机)。 我们用s[r]-s[l-1]表示[l,r]的区间和,看数据范围我们知道s[i]最多为1e6, 也就是说我们要的平方和最多不超过1e3,那样的话我们就可以试着枚举n以及平方和, 然后记录值等于s[r]j2s[r]-j^2的前缀和的个数即可得到答案。

代码

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