珂朵莉与宇宙 牛客

链接：
来源：牛客网
 

题目描述

星神是来自宇宙的

所以珂朵莉也是吧

所以我就出了个题

给你一个长为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

思路：涉及到区间求和的问题我们可以使用前缀和。由题意可得

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);
	}
}