完全平方数

[GESP202403 三级] 完全平方数

题目描述

小杨同学有一个包含 $n$ 个非负整数的序列 $A$，他想要知道其中有多少对下标组合 $<i,j>$（$1\le i<j\le n$），使得 $A_i+A_j$ 是完全平方数。

如果 $x$ 是完全平方数，则存在非负整数 $y$ 使得 $y\times y=x$。

输入格式

第一行一个非负整数 $n$，表示非负整数个数。
第二入行包含 $n$ 个非负整数 $A_1,A_2,\dots A_n$，表示序列 $A$ 包含的非负整数。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

5
1 4 3 3 5

样例输出 #1

3

提示说明

对全部的测试数据，保证 $1\le n\le 1000$，$0\le A_i\le 10^5$。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const ll N=1e4+10;
ll a[N];

bool cmb(ll x)
{
    for(ll i=0;;i++)
    {
        if(i*i==x)
            return 1;
        if(i*i>x)
            return 0;
    }
}

int main()
{
    ll n;
    cin>>n;

    for(ll i=1;i<=n;i++)
        cin>>a[i];

    ll res=0;
    for(ll i=1;i<=n;i++)
    {
        for(ll j=i+1;j<=n;j++)
        {
            ll x=a[i]+a[j];
            ll t=cmb(x);
            if(t) res++;
        }
    }

    cout<<res<<endl;
    return 0;
}