题目大意:给定n个数,如果两个数互质且平方和为完全平方数则不能同时被选,求选出一些数的最大和
首先这肯定是网络流无误 但是建图十分巧妙
很容易发现两个奇数不满足条件一 两个偶数不满足条件2
于是这是一个二分图 跑最小割即可
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 3030
#define S 0
#define T (n+1)
#define INF 0x3f3f3f3f
using namespace std;
struct abcd{
int to,f,next;
}table[1001001];
int head[M],tot=1;
int n,ans,a[M];
int dpt[M];
void Add(int x,int y,int z)
{
table[++tot].to=y;
table[tot].f=z;
table[tot].next=head[x];
head[x]=tot;
}
void Link(int x,int y,int z)
{
Add(x,y,z);
Add(y,x,0);
}
bool Check(int x,int y)
{
if(__gcd(x,y)^1) return false;
int temp=x*x+y*y;
if( int(sqrt(temp)+1e-7)*int(sqrt(temp)+1e-7)==temp )
return true;
return false;
}
bool BFS()
{
static int q[M];
int i,r=0,h=0;
memset(dpt,-1,sizeof dpt);
q[++r]=S;dpt[S]=1;
while(r!=h)
{
int x=q[++h];
for(i=head[x];i;i=table[i].next)
if(table[i].f&&!~dpt[table[i].to])
{
dpt[table[i].to]=dpt[x]+1;
q[++r]=table[i].to;
if(table[i].to==T)
return true;
}
}
return false;
}
int Dinic(int x,int flow)
{
int i,left=flow;
if(x==T) return flow;
for(i=head[x];i&&left;i=table[i].next)
if(table[i].f&&dpt[table[i].to]==dpt[x]+1)
{
int temp=Dinic(table[i].to,min(left,table[i].f) );
if(!temp) dpt[table[i].to]=-1;
left-=temp;
table[i].f-=temp;
table[i^1].f+=temp;
}
return flow-left;
}
int main()
{
int i,j;
cin>>n;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
ans+=a[i];
if(a[i]&1) Link(S,i,a[i]);
else Link(i,T,a[i]);
}
for(i=1;i<=n;i++)
if(a[i]&1)
for(j=1;j<=n;j++)
if(~a[j]&1)
if( Check(a[i],a[j]) )
Link(i,j,INF);
while( BFS() )
ans-=Dinic(S,INF);
cout<<ans<<endl;
return 0;
}