链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
题目描述
星神是来自宇宙的
所以珂朵莉也是吧
所以我就出了个题
给你一个长为n的序列a,有n*(n+1)/2个子区间,问这些子区间里面和为完全平方数的子区间个数
输入描述:
第一行一个数n 第二行n个数表示序列a
输出描述:
输出一个数表示答案
示例1
输入
复制6 0 1 0 9 1 0
6 0 1 0 9 1 0
输出
复制11
11
备注:
1 <= n <= 100000
0 <= ai <= 10
思路:涉及到区间求和的问题我们可以使用前缀和。由题意可得,那么我们只用遍历一遍前缀和数组,并记录r之前的各个区间数,找出满足的即可。
import java.util.Scanner;
public class Main {
static int N=100010;
static long ans=0;
static long cnt[]=new long[N*10];//记录区间和
static long s[]=new long[N];
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner cin=new Scanner(System.in);
int n=cin.nextInt();
for(int i=1;i<=n;i++){//求前缀和
int x=cin.nextInt();
s[i]=s[i-1]+x;
}
cnt[0]=1;
for(int i=1;i<=n;i++){
for(int j=0;j*j<=s[i];j++){
ans+=cnt[(int) (s[i]-(long)j*j)];
}
cnt[(int) s[i]]++;
}
System.out.println(ans);
}
}