Wannafly挑战赛5 A 前缀和+暴力

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 65536K，其他语言131072K
64bit IO Format: %lld

题目描述

星神是来自宇宙的

所以珂朵莉也是吧

所以我就出了个题

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

输入描述:

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

输出描述:

输出一个数表示答案

示例1

输入

6
0 1 0 9 1 0

输出

11

备注:

1 <= n <= 100000
0 <= ai <= 10

因为a给定范围,由此可知完全平方数最大为1e6,这枚举平方数1---1e3

求前缀和,枚举1---n 计算前面出现过多少个 sum[i]-i*i 

注意0需要初始化

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
 
 
 
int n;
int a[100015];
int ss[100015];
int ct[1000105];
int sum =0;
int main()
{
//    freopen("data.txt","r",stdin);
    ios::sync_with_stdio(false);
    cin >>n;
    for(int i=1;i<=n;i++)
    {
        cin >> a[i];
        ss[i]=ss[i-1]+a[i];
        sum+=a[i];
    }
    ll ans  = 0;
    for(int i=0;i<=1000&&i*i<=sum;i++)
    {
        memset(ct,0,sizeof(ct));
        ct[0]=1;
        int xx = i*i;
        for(int j=1;j<=n;j++)
        {
            if(ss[j]>=xx)
            {
                ans += ct[ ss[j] -xx ];
            }
            ct[ss[j]]++;
        }
    }
    cout << ans<<endl;
 
    return 0;
}