题目描述:
链接:https://ac.nowcoder.com/acm/problem/14600
来源:牛客网
星神是来自宇宙的
所以珂朵莉也是吧
所以我就出了个题
给你一个长为n的序列a,有n*(n+1)/2个子区间,问这些子区间里面和为完全平方数的子区间个数
思路:
前缀和记录,记录完你会发现枚举的话会超时。那怎么办呢?我们换一种枚举的方法,枚举平方数,因为观察可以发现ai的范围很小。而且因为是前缀和,那一定是单调不递减的,一遍就完事。
代码:
#include <cstdio>
#include <cstring>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <cmath>
#include <algorithm>
#include<bits/stdc++.h>
using namespace std;
const double N = 1e6+10;
const double pi = acos(-1.0);
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
const inline int read(){
int k = 0, f = 1; char c = getchar();
for(;!isdigit(c); c = getchar())
if(c == '-') f = -1;
for(;isdigit(c); c = getchar())
k = k * 10 + c - '0';
return k * f;
}
#define ll long long
#define CL(a,b) memset(a,b,sizeof(a))
#define MAXN 10000010
int a[MAXN];
int b[MAXN];
int main()
{
int n;
cin >> n;
long long int res = 0;
int k;
CL(b,0);
b[0] = 1;
a[0] = 0;
for(int i = 1 ; i <= n ; i++)
{
cin >> k;
a[i] = a[i-1]+k;
for(int j = 0 ; j*j <= a[i] ; j++)
{
if(b[a[i]-j*j]!=0)
{
res = res + b[a[i]-j*j] ;
}
}
b[a[i]]++;
}
cout << res << endl;
return 0;
}