題目大意:給定n個數,求出和最大的一個集合使得集合中任意兩個數要麼滿足平方和不爲平方數,要麼滿足有大於1公因數。
題解:可以將數字按照奇偶性分成兩半,因爲兩個奇數的平方和不可能爲平方數,兩個偶數有大於1的公因數,這樣不能共存的兩個數就位於左右兩邊,只要枚舉兩個數判斷一下,如果不能共存就在中間連上一條INF,奇數向左邊連費用,偶數向右邊連費用,就變成了一個最小割模型。
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<iomanip>
#include<cstring>
#include<string>
#include<algorithm>
#include<ctime>
#include<cmath>
using namespace std;
const int INF=2*1e9;
struct bian
{
int l,r,f;
}a[2000000];
int nex[2000000];
int fir[2000000];
int tot=1;
void _add_edge(int l,int r,int f)
{
a[++tot].l=l;
a[tot].r=r;
a[tot].f=f;
nex[tot]=fir[l];
fir[l]=tot;
}
void add_edge(int l,int r,int f)
{
_add_edge(l,r,f);
_add_edge(r,l,0);
}
int S=0,T=1001;
int d[1005];
bool bfs()
{
memset(d,-1,sizeof(d));
static int dui[1005];
int s=1,t=1;
dui[t++]=S;
d[S]=0;
while(s<t)
{
int u=dui[s];
s++;
for(int o=fir[u];o;o=nex[o])
{
if(!a[o].f) continue;
if(d[a[o].r]!=-1) continue;
d[a[o].r]=d[u]+1;
dui[t++]=a[o].r;
if(a[o].r==T) return true;
}
}
return false;
}
int dinic(int u,int flow)
{
if(u==T) return flow;
int left=flow;
for(int o=fir[u];o && left;o=nex[o])
{
if(a[o].f && d[a[o].r]==d[u]+1)
{
int temp=dinic(a[o].r,min(left,a[o].f));
a[o].f-=temp;
a[o^1].f+=temp;
left-=temp;
if(!temp) d[a[o].r]=-1;
}
}
return flow-left;
}
int x[2000];
int y[2000];
int gcd(int x,int y)
{
if(y==0) return x;
else return gcd(y,x%y);
}
int n;
int main()
{
scanf("%d",&n);
int ans=0;
for(int i=1;i<=n;i++) scanf("%d",&x[i]);
for(int i=1;i<=n;i++) scanf("%d",&y[i]),ans+=y[i];
for(int i=1;i<=n;i++)
{
if(x[i]&1) add_edge(S,i,y[i]);
else add_edge(i,T,y[i]);
}
for(int i=1;i<=n;i++)
{
if(!(x[i]&1)) continue;
for(int j=1;j<=n;j++)
{
if((x[j]&1)) continue;
int t=gcd(x[i],x[j]);
long long mid=1ll*x[i]*x[i]+1ll*x[j]*x[j];
long long tt=sqrt(mid);
if(tt*tt==mid && t==1) add_edge(i,j,INF);
}
}
while(bfs()) ans-=dinic(S,INF);
cout<<ans;
return 0;
}