前缀和转化

链接：https://ac.nowcoder.com/acm/contest/36/A?&headNav=www
 

给你一个长为n的序列a，有n*(n+1)/2个子区间，问这些子区间里面和为完全平方数的子区间个数

输入描述:

第一行一个数n
第二行n个数表示序列a

输出描述:

输出一个数表示答案

示例1

输入

复制

6
0 1 0 9 1 0

输出

复制

11

备注:

1 <= n <= 100000

0 <= ai <= 10

#include<cstdio>
#include<cstring>
using namespace std;
int a[100005];
int sum[100005];
int pos[1000005];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i];
    long long ans=0;
    pos[0]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j*j<=sum[i];j++)//枚举平方数区间来优化
            ans+=pos[sum[i]-j*j];
        pos[sum[i]]++;
    }
    printf("%lld\n",ans);
}