wannfly 挑战赛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

考虑综合不超过1e6，那么每次枚举这个平方和是否出现过

#include <bits/stdc++.h>
using namespace std;
 
int mp[10000010];
 
int main()
{
    int n;
    scanf("%d",&n);
    int x;
    int sum=0;
    long long ans=0;
    mp[0]=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        sum+=x;
        for(int j=0;j<=1000&&(j*j<=sum);j++)
            ans+=mp[sum-j*j];
        mp[sum]++;
    }
    printf("%lld\n",ans );
 
}