Wannafly挑战赛5 A.珂朵莉与宇宙

Wannafly挑战赛5 A.珂朵莉与宇宙
题目描述

星神是来自宇宙的 

所以珂朵莉也是吧 

所以我就出了个题 

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

输入描述:
第一行一个数n
第二行n个数表示序列a

输出描述:
输出一个数表示答案
 

示例1 


输入

6
0 1 0 9 1 0


#include<cstdio>
using namespace std;
int sum[100010];
int a[1000100];
int main ()
{ 
    int n;
    long long ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
     scanf("%d",&sum[i]);
     sum[i]+=sum[i-1];
     //a[sum[i]]++;先把a数组算出来是不对的
    }
    /*分为两种情况：
    1是第一个数是完全平方数
    2是第一个数不是完全平方数
    所以a[0]=1就是为了处理第一种情况的
    如果是第二种情况就没必要让a[0]=1了*/
    a[0]=1;
    for(int i=1;i<=n;i++){
     for(int j=0;j*j<=sum[i];j++){
      ans+=a[sum[i]-j*j];
     }
     //a[sum[i]]++表示某个前缀和出现的次数,且不能先把a数组给算出来，
    a[sum[i]]++;
    }
    printf("%lld\n", ans);
    return 0;
}


前缀和a[j]肯定是大于等于可能满足的平方数的，所以前面用求得的n个前缀和   依次去 减去 一个平方数  得到的数   就是要截去的数目（也即一个前缀和），因为区间是连续的，所以只能是减掉某一个前缀和，所以就转化成求某些前缀和的总个数了

由题意也可得：sum[m]  -  sum[n]  =  j  *   j   ,并且 m>=n  
j  就是那个平方数 ，并且 j 在 闭区间[n，m] 内。由此可以看出求  j 的个数 就等价于就  sum[n]  的个数了  ( 有点逆向思维的味道)