珂朵莉与宇宙（前缀和）

珂朵莉与宇宙

星神是来自宇宙的

所以珂朵莉也是吧

所以我就出了个题

给你一个长为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<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e6+10;
int n;
int a[N],s[N];
int cnt[N];
int ans;

signed main(){
    cin>>n;
    cnt[0] = 1; // 初始化，空子区间的情况
    for(int i=1;i<=n;i++){
        cin>>a[i];
        s[i]=s[i-1]+a[i];
        
        // 枚举可能的完全平方数
        for(int j=0;j*j<=s[i];j++){
            int target = s[i] - j*j;
            if(target >=0){
                ans += cnt[target];
            }
        }
        
        cnt[s[i]]++; // 将当前前缀和加入计数
    }
    cout<<ans;
    return 0;
}