原题链接:登录—专业IT笔试面试备考平台_牛客网
解题思路:首先用记录原数组的前缀和,然后从第一位开始遍历。对于每一个前缀和的值,尝试所有小于其的完全平方数。如果这个前缀和的值减去该完全平方数的差在之前的前缀和出现过,则说明之前出现的位置到当前位置的这一段区间内是一个完全平方数。
如果之前出现过多次,那么说明也有多个起点到当前位置组成的多个区间符合条件。记录下区间数,累加进答案中。然后往记录之前前缀和的每个值出现过几次的数组中,记录当前位置的值所对应次数加一即可。
AC代码:
#include<bits/stdc++.h>
using namespace std;
long long n,a;
long long pre_sum[1000010];//pre_sum为原数组的前缀和
long long cmp_ps[10000100];//记录前缀和中每个数数出现的次数
long long ans=0;//答案数
int main(){
cin>>n;
for(int i=1;i<=n;i++){
scanf("%lld",&a);
pre_sum[i]=pre_sum[i-1]+a;
}
cmp_ps[0]=1;
for(int i=1;i<=n;i++){
for(int j=0;j*j<=pre_sum[i];j++) ans+=cmp_ps[pre_sum[i]-j*j];//如果cmp_ps不为0,说明之前的前缀和出现过这个数字,值就是出现的次数。
//剪掉前缀和为pre_sum[i]-j*j的那一段,就说明中间一段可以被j*j整除,而前面出现过多次该数,说明就有多个选择
cmp_ps[pre_sum[i]]++;//记录该值出现过1次
}
cout<<ans;
return 0;
}