BZOJ3275 Number-最小割

28 篇文章 0 订阅
20 篇文章 0 订阅

题目链接:右转进入题目

这个题有一部非常好的转化。

首先一眼看出这是一个带权最大独立团的问题。

所以想到网络流

但是网络流里面学的带权最大独立团问题必须要求是二分图啊怎么办!

我承认我没有想到怎么二分QwQ真是太弱辣!

发现两个奇数不满足a^2+b^2=c^2的条件(需要一点数学知识),两个偶数不满足gcd(a,b)=1的条件。

所以分别把奇数和偶数看作两排点即可。

如果两个不能同时选取,就连上inf的边。

s和奇数,偶数和t之间连边边权为数的权值。

所有数字的权值之和 - 最小割就是答案。

代码:

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<queue>
#include<climits>
#define INF INT_MAX
#define MAXN 3010
#define MAXM 5000000
using namespace std;
struct edges{
	int to,pre,resf;
}e[MAXM];int etop,a[MAXN],b[MAXN];
int h[MAXN],cur[MAXN],lev[MAXN];
bool vis[MAXN];queue<int> q;
bool bfs(int s,int t)
{
	memset(vis,false,sizeof(vis));
	memset(lev,0,sizeof(lev));
	while(!q.empty()) q.pop();
	q.push(s);lev[s]=0;vis[s]=true;
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(int i=h[x];i;i=e[i].pre)
			if(!vis[e[i].to]&&e[i].resf)
			{
				vis[e[i].to]=true;
				lev[e[i].to]=lev[x]+1;
				q.push(e[i].to);
			}
	}
	return vis[t];
}
int dfs(int s,int t,int a)
{
	if(s==t||!a) return a;
	int flow=0,f;
	for(int &i=cur[s];i;i=e[i].pre)
		if(lev[e[i].to]==lev[s]+1&&(f=dfs(e[i].to,t,min(a,e[i].resf)))>0)
		{
			flow+=f;a-=f;e[i].resf-=f;
			e[((i-1)^1)+1].resf+=f;
			if(!a) break;
		}
	return flow;
}
int add_edge(int u,int v,int c)
{
	etop++;
	e[etop].to=v;
	e[etop].pre=h[u];
	h[u]=etop;
	e[etop].resf=c;
	return 0;
}
int gcd(int a,int b)
{
	return a?gcd(b%a,a):b;
}
bool check(int a,int b)
{
	int c=sqrt(a*a+b*b+0.5);
	return (c*c==a*a+b*b)&&(gcd(a,b)==1);
}
int main()
{
	int n,n1,n2,totans=0;scanf("%d",&n);
	int s,t,cnt;s=0,t=n+1,n1=0,n2=n+1;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++) totans+=a[i];
	for(int i=1;i<=n;i++)
		if(a[i]&1) b[++n1]=a[i];
		else b[--n2]=a[i];
	for(int i=1;i<=n1;i++)
		for(int j=n1+1;j<=n;j++)
			if(check(b[i],b[j]))
				add_edge(i,j,INF),add_edge(j,i,0);
	for(int i=1;i<=n1;i++)
		add_edge(s,i,b[i]),add_edge(i,s,0);
	for(int i=n1+1;i<=n;i++)
		add_edge(i,t,b[i]),add_edge(t,i,0);
	int flow=0;
	while(bfs(s,t))
	{
		for(int i=s;i<=t;i++)
			cur[i]=h[i];
		flow+=dfs(s,t,INF);
	}
	printf("%d\n",totans-flow);
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值