牛客网Wannafly挑战赛27 A： 灰魔法师

牛客网Wannafly挑战赛27 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 <= 105)
第二行 n 个整数 ai (1 <= ai <= 105)

输出描述:

输出一个整数，表示满足上述条件的数对个数。

示例1

输入

复制

3
1 3 6

输出

复制

2

说明

满足条件的有 (1, 2), (2, 3) 两对。

#include<stdio.h>
#include<math.h>
#include<string.h>
#define N 110050
int a[2*N];
int main()
{int n,i,j,num;long long sum=0;while(scanf("%d",&n)!=EOF){sum=0;for(i=0;i<n;i++){scanf("%d",&num);for(j=1;j<(int)sqrt(2*N);j++){if(j*j-num>0){sum+=a[j*j-num];}		}a[num]++;}printf("%lldn",sum);}return 0;
}