BZOJ 3275: Number|网络流

网络流

奇偶分组 排斥的连边保证不同时选

具体：

S->奇数    流量为奇数大小

偶数->T    流量为偶数大小

奇偶若有排斥  奇数->偶数  流量为无穷大

求一个最小割，最后割断了哪个边，那个边上的数就不选

#include<set>
#include<map>
#include<ctime>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define inf 1e9
using namespace std;
int dis[3333],q[3333];
int head[3333],lst[3333333],nxt[3333333],v[3333333];
int a[3333],b[3333],n,tot=1,a1,b1;
int S,W,sum;
void insert(int x,int y,int z)
{
//cout << x <<" "<<y <<" "<< z<<endl;
lst[++tot]=y; nxt[tot]=head[x]; v[tot]=z; head[x]=tot;
lst[++tot]=x; nxt[tot]=head[y]; v[tot]=0; head[y]=tot;
}
int gcd(int x,int y){return x==0?y:gcd(y%x,x);}
bool BFS()
{
memset(dis,0,sizeof(dis)); dis[S]=1;
int l=1,r=2; q[1]=S;
while(l<r)
{
int x=q[l++];
for(int i=head[x];i;i=nxt[i])
if(v[i]&&dis[lst[i]]==0)
{
dis[lst[i]]=dis[x]+1;
q[r++]=lst[i];
}
}
return dis[W];
}
int dfs(int x,int f)
{
if(x==W)return f;
int ww=0,w;
for(int i=head[x];i;i=nxt[i])
if(v[i]&&dis[lst[i]]==dis[x]+1)
{
w=dfs(lst[i],min(f-ww,v[i]));
ww+=w; v[i]-=w; v[i^1]+=w;
if(ww==f)return f;
}
if(W==0)dis[x]=0;
return ww;
}
int main()
{
scanf("%d",&n);
S=n+1; W=S+1;
for(int i=1;i<=n;i++)
{
int x;scanf("%d",&x);
if(x&1) a[++a1]=x;
else b[++b1]=x;
sum+=x;
}
for(int i=1;i<=a1;i++)
for(int j=1;j<=b1;j++)
{
int x=a[i]*a[i]+b[j]*b[j];
int y=sqrt(x);
if(y*y==x&&gcd(a[i],b[j])==1) insert(i,a1+j,inf);
}
for(int i=1;i<=a1;i++)insert(S,i,a[i]);
for(int i=1;i<=b1;i++)insert(i+a1,W,b[i]);
while(BFS())sum-=dfs(S,inf);
cout << sum;
return 0;
}