勾股定理

勾股定理

Time Limit: 1000ms Memory limit: 65536K 有疑问？点这里^_^

题目描述

给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

提示

来源

示例程序

#include <string.h>
#include <stdio.h>
#include<stdlib.h>
int cmp(const void *a,const void *b)
{
return *(int *)a-*(int *)b;
}
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];
}
qsort(a,n,sizeof(a[0]),cmp);
qsort(ls,n,sizeof(ls[0]),cmp);
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;
}