BZOJ 3275 Number 最小割

标签：bzoj   bzoj3275   最小割   dinic   

题目大意：给定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;
}

BZOJ 3275 Number 最小割

标签：bzoj   bzoj3275   最小割   dinic   

原文地址：http://blog.csdn.net/popoqqq/article/details/41894949