Wannafly挑战赛27 A题-灰魔法师（Java版）

最新推荐文章于 2020-05-19 22:50:56 发布

仰望星空的程序缘

最新推荐文章于 2020-05-19 22:50:56 发布

阅读量200

点赞数

分类专栏：牛客文章标签：牛客算法

本文链接：https://blog.csdn.net/weixin_42977872/article/details/83830846

版权

牛客专栏收录该内容

15 篇文章 0 订阅

订阅专栏

链接：https://ac.nowcoder.com/acm/contest/215/A
来源：牛客网

题目描述
“White shores, and beyond. A far green country under a swift sunrise.”–灰魔法师

给出长度为n的序列a, 求有多少对数对 (i, j) (1 <= i < j <= n) 满足 ai + aj 为完全平方数。
输入描述:
第一行一个整数 n (1 <= n <= 10^5)
第二行 n 个整数 ai (1 <= ai <= 10^5)
输出描述:
输出一个整数，表示满足上述条件的数对个数。
示例1
输入
3
1 3 6
输出
2
说明
满足条件的有 (1, 2), (2, 3) 两对。

因为ai最大为10^5，所以最大是20万，也就是447的平方，然后就可以暴力了

import java.util.Scanner;
//447
public class Main {
	
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		long cnt=0;
		int a[]=new int[200010];
		int n=sc.nextInt();
		for(int i=1;i<=n;i++){
			int k=sc.nextInt();
			for(int j=2;j<=447;j++){
				if(j*j>k)
					cnt+=a[j*j-k];
			}
			a[k]++;
		}
		System.out.println(cnt);
		sc.close();		
				
	}
}