灰魔法师

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

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

“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<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=300000+100;
int a[maxn];
int zhi[maxn];
int main()
{
    int n,x;
    while(scanf("%d",&n)!=EOF){
            ///freopen("D:\\in.txt","r",stdin); //输入重定向，输入数据将从D盘根目录下的in.txt文件中读取
    ///freopen("ouut.txt","w",stdout); //输出重定向，输出数据将保存在D盘根目录下的out.txt文件中
 
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++){
            scanf("%d",&x);
            zhi[i]=x;
            a[x]++;
        }
        LL sum=0;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=500;j++){
                if(j*j>zhi[i])
                {
                    if(j*j-zhi[i]==zhi[i])
                        sum=sum+a[j*j-zhi[i]]-1;
                    else
                        sum+=a[j*j-zhi[i]];
              ///  if(a[j*j-zhi[i]]>0)
                 /// cout<<"1:   "<<i<<"   "<<j<<"  "<<"   "<<j*j-zhi[i]<<"***"<<a[j*j-zhi[i]]<<endl;
                }
                /*
                else
                {
                    if(zhi[i]-j*j==zhi[i])
                        sum=sum+a[zhi[i]-j*j]-1;
                    else
                        sum+=a[zhi[i]-j*j];
                   if(a[zhi[i]-j*j]>0)
                    cout<<"2:"<<i<<"   "<<j<<"   "<<zhi[i]-j*j<<"***"<<a[zhi[i]-j*j]<<endl;
                }
            }
            */
        }
    }
        printf("%lld\n",sum/2);
    }
    return 0;
}