BZOJ 3275: Number|网络流

最新推荐文章于 2019-07-04 22:57:47 发布

ws_yzy

最新推荐文章于 2019-07-04 22:57:47 发布

阅读量408

点赞数

分类专栏：网络流文章标签：网络流

本文链接：https://blog.csdn.net/ws_yzy/article/details/50532153

版权

网络流专栏收录该内容

23 篇文章 0 订阅

订阅专栏

网络流

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

具体：

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;
}