BZOJ 3158: 千钧一发 最小割

3158: 千钧一发

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1463  Solved: 540
[Submit][Status][Discuss]

Description

Input

第一行一个正整数N。

第二行共包括N个正整数，第 个正整数表示Ai。

第三行共包括N个正整数，第 个正整数表示Bi。

Output

共一行，包括一个正整数，表示在合法的选择条件下，可以获得的能量值总和的最大值。

Sample Input

4
3 4 5 12
9 8 30 9

Sample Output

39

HINT

1<=N<=1000,1<=Ai,Bi<=10^6

最小割

同3275: Number 传送门

#include<cmath>#include<ctime>#include<cstdio>#include<cstdlib>#include<cstring>#include<complex>#include<iostream>#include<algorithm>#include<iomanip>#include<vector>#include<string>#include<queue>#include<set>#include<map>using namespace std;typedef long long ll;inline ll read(){ll x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return f*x;}const int N=3100;const ll inf=0X7f7f7f7f7f7f7fll;int last[N],ecnt=1,even[N],odd[N],q[N],b[N],d[N],S=0,T=N-1;ll ans,a[N],n;struct EDGE{int to,nt;ll val;}e[1200010];inline void readd(int u,int v,ll val){e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}inline void add(int u,int v,ll val){readd(u,v,val);readd(v,u,0);}int gcd(int x,int y){return y==0?x:gcd(y,x%y);}bool bfs(){memset(d,0,sizeof(d));int head=0,tail=1;d[S]=1;q[head]=S;while(head<tail){int u=q[head++];for(int i=last[u];i;i=e[i].nt)if(e[i].val&&!d[e[i].to]){d[e[i].to]=d[u]+1;q[tail++]=e[i].to;}}return d[T];}ll dfs(int u,ll lim){if(u==T||!lim)return lim;ll flow=0;for(int i=last[u];i;i=e[i].nt)if(d[e[i].to]==d[u]+1){ll tmp=dfs(e[i].to,min(lim,e[i].val));e[i].val-=tmp;e[i^1].val+=tmp;lim-=tmp;flow+=tmp;if(!lim)break;}if(!flow)d[u]=-1;return flow;}void dinic(){while(bfs())ans+=dfs(S,inf);}int main(){ll n=read();ll sum=0;for(int i=1;i<=n;i++)a[i]=read();for(int i=1;i<=n;i++)b[i]=read();for(int i=1;i<=n;i++){sum+=b[i];if(a[i]&1)add(S,i,b[i]),odd[++odd[0]]=i;else add(i,T,b[i]),even[++even[0]]=i;}for(int i=1;i<=odd[0];i++)for(int j=1;j<=even[0];j++){ll tmp=a[odd[i]]*a[odd[i]]+a[even[j]]*a[even[j]];ll t=sqrt(tmp);if(t*t==tmp&&gcd(a[odd[i]],a[even[j]])==1)add(odd[i],even[j],inf);}dinic();printf("%lld\n",sum-ans);return 0;} /*43 4 5 129 8 30 9Sample Output39*/