Elliot_Lin_洛谷B3957 [GESP202403 三级] 完全平方数（C语言）

[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<stdio.h>
#include<math.h>
#define N 1000
int main(){
    int n, a[N], count = 0;
    scanf("%d", &n);
    for(int i = 0; i < n; i++){
        scanf("%d", &a[i]);
    }
    // 判断
    for(int i = 0; i < n; i++){
        for(int j = i+1; j < n; j++){
            // 如果不是平方数，则浮点数x 强制转化为 整数
            int x = (int)sqrt(a[i]+a[j]);
            if(x*x == a[i] + a[j])
                count++; 
        }
    }
    printf("%d", count);
    return 0;
}