珂朵莉与宇宙

链接：https://ac.nowcoder.com/acm/problem/14600
来源：牛客网

题目描述
星神是来自宇宙的

所以珂朵莉也是吧

所以我就出了个题

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

链接：https://ac.nowcoder.com/acm/problem/14600
来源：牛客网

输入描述:
第一行一个数n
第二行n个数表示序列a
输出描述:
输出一个数表示答案
示例1
输入
复制
6
0 1 0 9 1 0
输出
复制
11
备注:
1 <= n <= 100000

0 <= ai <= 10

这是一道非常巧妙的思维题
直接暴力5e9一定会TLE，因此得想办法改进时间复杂度。
观察备注，发现最多到1000的平方，因为我们在依次枚举前缀和的时候可以统计前缀和对应的数量，然后枚举平方项即可：
$$s[r]-s[l-1]=k^2$$
将式子变形为
$$s[r]-k^2=s[l-1]$$
因此我们只需要枚举r的时候枚举k（1~1000）顺便统计一下个数即可

注意别忘了开long long

#include<bits/stdc++.h>
using namespace std;
#define N 100010

#define int long long

int n,a[N];
int cnt[N*10];

signed main(){
    scanf("%lld",&n);
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        a[i]+=a[i-1];
    }
    
    int ans=0;
    for(int i=0;i<=n;i++){
        for(int j=0;j*j<=a[i]&&j<=1000;j++){
            ans+=cnt[a[i]-j*j];
        }
        cnt[a[i]]++;
    }
    
    cout << ans << endl;
    
    return 0;
}