BZOJ 3158 千钧一发 最小割

 

分析：

　　偶数对满足条件2，所有奇数对满足条件1。

　　如果你能一眼看出这个规律，这道题就完成了一半。

　　我们只需要将数分为两类，a值为奇数，就从S向这个点连容量为b值的边，a值为偶数，就从这个点向T连容量为b值的边。

　　暴力枚举，对于奇集合和偶集合中不能共存的两个数，连容量为无穷大的边。

　　求出最小割，代表这个割集要被我们舍弃。

　　然后直接用b值总和减去最小割就好。

代码：

#include<bits/stdc++.h>
#define ms(a,x) memset(a,x,sizeof(a))
#define pf(x) (1LL*x*x)
#define ll long long
using namespace std;
const int N=1005,M=500005,inf=0x3f3f3f3f;
struct node{int y,z,nxt;}e[M*2];int S,T,tot=0;
int d[N],h[N],c=1,a[N],b[N],q[N],n,m;ll s[N];
int gcd(int x,int y){
    return y?gcd(y,x%y):x;
} bool pd(ll x){
    ll y=sqrt(x);return y*y!=x;
} void add(int x,int y,int z){
    e[++c]=(node){y,z,h[x]};h[x]=c;
    e[++c]=(node){x,0,h[y]};h[y]=c;
} bool bfs(){
    int f=1,t=0;ms(d,-1);
    q[++t]=S;d[S]=0;
    while(f<=t){
        int x=q[f++];
        for(int i=h[x],y;i;i=e[i].nxt)
        if(d[y=e[i].y]==-1&&e[i].z)
        d[y]=d[x]+1,q[++t]=y;
    } return (d[T]!=-1);
} int dfs(int x,int f){
    if(x==T) return f;int w,tmp=0;
    for(int i=h[x],y;i;i=e[i].nxt)
    if(d[y=e[i].y]==d[x]+1&&e[i].z){
        w=dfs(y,min(e[i].z,f-tmp));
        if(!w) d[y]=-1;
        e[i].z-=w;e[i^1].z+=w;tmp+=w;
        if(tmp==f) return f;
    } return tmp;
} void dinic(){
    while(bfs()) tot+=dfs(S,inf);
} signed main(){
    scanf("%d",&n);int ans=0;S=0;T=n+1;
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]),s[i]=pf(a[i]);
    for(int i=1;i<=n;i++)
    scanf("%d",&b[i]),ans+=b[i];
    for(int i=1;i<=n;i++)
    for(int j=i+1;j<=n;j++)
    if(gcd(a[i],a[j])==1&&!pd(s[i]+s[j]))
    a[i]&1?add(i,j,inf):add(j,i,inf);
    for(int i=1;i<=n;i++)
    a[i]&1?add(S,i,b[i]):add(i,T,b[i]);
    dinic();printf("%d\n",ans-tot);
    return 0;
}