[海豹海边爆]牛客14600

原题链接:登录—专业I​​​​​​T笔试面试备考平台_牛客网

解题思路:首先用记录原数组的前缀和,然后从第一位开始遍历。对于每一个前缀和的值,尝试所有小于其的完全平方数。如果这个前缀和的值减去该完全平方数的差在之前的前缀和出现过,则说明之前出现的位置到当前位置的这一段区间内是一个完全平方数。

如果之前出现过多次,那么说明也有多个起点到当前位置组成的多个区间符合条件。记录下区间数,累加进答案中。然后往记录之前前缀和的每个值出现过几次的数组中,记录当前位置的值所对应次数加一即可。

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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值