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

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

#include
#include
#include
#include
#includeusing 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;
}


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部