要先判断a*a+b*b==c*c这个条件.再判断gcd(a,b)==1.不然会TLE.
/* I will wait for you */
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iostream>
#include<fstream>
#include<vector>
#include<queue>
#include<deque>
#include<set>
#include<map>
#include<string>
#define make(a,b) make_pair(a,b)
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
const int maxn=6010;
const int maxm=1000010;
const int maxs=26;
const int INF=1<<29;
const int P=1000000007;
const double error=1e-9;
struct edge{ int v,a,next; }e[maxm];
int s,t,ans,cnt,head[maxn],dis[maxn],cur[maxn],q[maxn],c[maxn];
int gcd(int a,int b)
{
return a?gcd(b%a,a):b;
}
int allow(int a,int b)
{
int c=sqrt(a*a+b*b);
return c*c==(a*a+b*b)&&gcd(a,b)==1;
}
void insert(int u,int v,int a)
{
e[cnt]=(edge){v,a,head[u]},head[u]=cnt++;
e[cnt]=(edge){u,0,head[v]},head[v]=cnt++;
}
int bfs()
{
memset(dis,-1,sizeof(dis));
int he=0,ta=1;q[0]=s,dis[s]=0;
while(he!=ta)
{
int u=q[he++];
for(int i=head[u],v;i!=-1;i=e[i].next)
if(e[i].a&&dis[v=e[i].v]==-1) dis[v]=dis[u]+1,q[ta++]=v;
}
return dis[t]!=-1;
}
int dfs(int u,int a)
{
if(u==t) return a;
int f,flow=0;
for(int i=cur[u],v;a&&i!=-1;i=e[i].next)
{
if(e[i].a&&dis[v=e[i].v]==dis[u]+1&&(f=dfs(v,min(a,e[i].a)))>0)
{
e[i].a-=f,e[i^1].a+=f,a-=f,flow+=f;
if(e[i].a) cur[u]=i;
}
}
if(!flow) dis[u]=-1;
return flow;
}
void dinic()
{
while(bfs()) memcpy(cur,head,sizeof(cur)),ans-=dfs(s,INF);
}
int main()
{
int n;scanf("%d",&n);
s=2*n+1,t=2*n+2;
memset(head,-1,sizeof(head));
for(int i=1;i<=n;i++)
{
scanf("%d",&c[i]);ans+=2*c[i];
insert(s,i,c[i]),insert(n+i,t,c[i]);
}
for(int i=1;i<=n;i++) for(int j=1;j<=n;j++)
if(allow(c[i],c[j])) insert(i,n+j,INF),insert(j,n+i,INF);
dinic();printf("%d\n",ans/2);
return 0;
}