珂朵莉与宇宙

#include<bits/stdc++.h>
const int N = 1e6 + 10;
int Cnt[N], S[N];
using namespace std;

int main()
{
    int n, x = 0;
    long long ans = 0;
    
    Cnt[0] = 1;//第一个数初始化 
    scanf("%d", &n);
   
    for(int i = 1; i <= n; i++ ){
        scanf("%d", &x);
        S[i] = S[i - 1] + x; //预处理前缀和
        
        for(int j = 0; j * j <= S[i]; j++ )
                ans += Cnt[S[i] - j * j];
        Cnt[S[i]] ++;// 更新前缀和出现的次数
    }
    
    printf("%lld\n", ans);
    return 0;
}