珂朵莉与宇宙

题目描述:
链接:https://ac.nowcoder.com/acm/problem/14600
来源:牛客网

星神是来自宇宙的

所以珂朵莉也是吧

所以我就出了个题

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

思路:
前缀和记录,记录完你会发现枚举的话会超时。那怎么办呢?我们换一种枚举的方法,枚举平方数,因为观察可以发现ai的范围很小。而且因为是前缀和,那一定是单调不递减的,一遍就完事。

代码:

#include <cstdio>
#include <cstring>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <cmath>
#include <algorithm>
#include<bits/stdc++.h>
using namespace std;
const double N = 1e6+10;
const double pi = acos(-1.0);
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
const inline int read(){
    int k = 0, f = 1; char c = getchar();
    for(;!isdigit(c); c = getchar())
        if(c == '-') f = -1;
    for(;isdigit(c); c = getchar())
        k = k * 10 + c - '0';
    return k * f;
}
#define ll long long
#define CL(a,b) memset(a,b,sizeof(a))
#define MAXN 10000010
int a[MAXN];
int b[MAXN];
int main()
{
    int n;
    cin >> n;
    long long int res = 0;
    int k;
    CL(b,0);
    b[0] = 1;
    a[0] = 0;
    for(int i = 1 ; i <= n ; i++)
    {
    	cin >> k;
    	a[i] = a[i-1]+k;
    	for(int j = 0 ; j*j <= a[i] ; j++)
    	{
    		if(b[a[i]-j*j]!=0)
    		{
    			res = res + b[a[i]-j*j] ;
			}
		}
		b[a[i]]++;
	}
	cout << res << endl;
	return 0;
}
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页