BZOJ3158 千钧一发-最小割

题目链接：右转进入题目

和[Number]这道题几乎一模一样。

不多解释

代码就是随便改了改

//BZOJ3275 
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<queue>
#include<climits>
#define INF INT_MAX
#define MAXN 1010
#define MAXM 3000000
#define ull long long
using namespace std;
struct edges{
	int to,pre,resf;
}e[MAXM];int etop,a[MAXN],b[MAXN],c[MAXN];
int h[MAXN],cur[MAXN],lev[MAXN],d[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((ull)a*a+(ull)b*b+0.5);
	return ((ull)c*c==(ull)a*a+(ull)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++) scanf("%d",&c[i]);
	for(int i=1;i<=n;i++) totans+=c[i];
	for(int i=1;i<=n;i++)
		if(a[i]&1) n1++,b[n1]=a[i],d[n1]=c[i];
		else n2--,b[n2]=a[i],d[n2]=c[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,d[i]),add_edge(i,s,0);
	for(int i=n1+1;i<=n;i++)
		add_edge(i,t,d[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;
}