牛客网+灰魔法师

题目描述

“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)

输出描述

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

输入

3
1 3 6

输出

2

说明

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

代码：

#include <bits/stdc++.h>
using namespace std;
int a[250005];
///动态规划的思想。。。。。
///我好菜啊。。。。
int main()
{
    int n,m;
    long long s = 0;
    scanf("%d",&n);
    for(int i=0; i<n; i++)
    {
        scanf("%d",&m);
        for(int j=1; j<=500; j++)///最大才是200000；所以找里面的平方数很好找。
        {
            if(j*j>m)///找互补的数的个数。
            {
                s = s + a[j*j-m];
            }
        }
        a[m]++;///数字个数时刻变化。
    }
    printf("%lld\n",s);
    return 0;
}