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

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 65536K，其他语言131072K
64bit IO Format: %lld

题目描述

星神是来自宇宙的

所以珂朵莉也是吧

所以我就出了个题

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

输入描述:

第一行一个数n
第二行n个数表示序列a

输出描述:

输出一个数表示答案

思路:

两个点:

1.子区间必然连续

2.整数的平方都是完全平方数,所以负数也是

 

一个前缀和,必然能通过减另一个前缀和来求出,也就是当前前缀和为a,减了了b,等于c,c是一个完全平方数,也就是a-b = c;

法诺来也就是a-c = b,我们只需要找这个减的b是否存在于之前的前缀和中.

 

然后这里有一个我才知道的问题是数组的访问速度比map快...

map快在插入和删除,但是改变值比起数组慢...毕竟有下标...

#include<bits/stdc++.h>
using namespace std;
const int maxn = 100003;
typedef long long ll;
int num[maxn];
int ditu[maxn*10];
int main()
{
    //freopen("in.txt","r",stdin);
    ios::sync_with_stdio(false);
    int n;
    cin>>n;
    for(int i=0; i<n; i++)
    {
        cin>>num[i];
        if(i == 0);
        else
            num[i] += num[i-1];
    }
    ditu[0] = 1;
    ll ans = 0;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j*j<=num[i]; j++)
        {
            ans += ditu[ num[i]-j*j ];
        }
        ditu[num[i]]++;
    }
    cout<<ans<<endl;
    return 0;
}