玛朵莉与宇宙暴力枚举+前缀和

岁余冬雨

于 2020-08-13 20:22:29 发布

阅读量628

点赞数

本文链接：https://blog.youkuaiyun.com/fusu123456789/article/details/107990587

版权

本文介绍了一种算法，用于解决数据范围内的完全平方数前缀和计数问题。通过枚举前缀和并统计小于当前前缀和的所有完全平方数，使用额外数组记录这些计数，实现高效求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：看题目数据范围，每个ai都是小于10的，那么直接暴力枚举前缀和的话是1e6的，不会超时，因此维护一个前缀和数组，对每个前缀和，枚举小于他的完全平方的个数，再用一个数组维护前缀和的完全平方的个数的值就可以了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;

const int N=1e6+10;
int a[N],m[N];

int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i],a[i]+=a[i-1];
    long long cnt=0;
    m[0]=1;
    for(int i=1;i<=n;i++){
        for(int j=0;j*j<=a[i];j++){
            cnt+=m[a[i]-j*j];
        }
        m[a[i]]++;
    }
    cout<<cnt<<endl;
    return 0;
}