勾股定理

题目描述

给N个数，判断这N个数中存在多少组勾股数（勾股数：存在三个数满足a*a + b*b = c*c）。

输入

第一行输入一个数T(1<=T<=50)，表示有T组数据。

每组数据输入一个N(1<=N<=1000)。

接下来N个数ai(1<=ai<=10^9)。

输出

每组数据输出一行，表示有多少组勾股数。

示例输入

3
4
4 3 5 3
4 
6 7 8 10
5
6 7 8 9 5

示例输出

2
1
0

这个题暴力的话会TLE，但是加一个二分就可以A掉

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<bits/stdc++.h>
using namespace std;
int  a[1010], b, ls[1010];
 int f(int  x, int  y)
 {
     int  m;
     while(x < y)
     {
         m = x +(y-x) /2;
         if(ls[m] == b) return 1;
         else if (ls[m] > b) y = m;
         else x = m+1;
     }
     return -1;
 }
int main()
{

    int t;
   while(~scanf("%d",&t))
    while(t--)
    {
        int n, i, j;
        scanf("%d",&n);
        for(i = 0; i < n; i++)
        {
            scanf("%d",&a[i]);
            ls[i] = a[i]*a[i];
        }
        sort(a,a+n);
        sort(ls,ls+n);
        int c = 0;
         for(i = 0; i < n; i++)
         {
             for(j = i+1; j < n; j++)
             {
                 int k;
                 b = a[i] *a[i] + a[j] *a[j];
                 k = f(j,n);
                 if(k == 1)
                  c+=k;
             }
         }
            printf("%d\n",c);
    }
    return 0;
}