BZOJ4514: [Sdoi2016]数字配对【二分图性质，贪心费用流】

题目描述：

有 n 种数字，第 i 种数字是 ai、有 bi 个，权值是 ci。
若两个数字 ai、aj 满足，ai 是 aj 的倍数，且 ai/aj 是一个质数，
那么这两个数字可以配对，并获得 ci×cj 的价值。
一个数字只能参与一次配对，可以不参与配对。
在获得的价值总和不小于 0 的前提下，求最多进行多少次配对。
n≤200，0≤ai≤10⁹，0≤bi≤10⁵，∣ci∣≤10^5

题目分析：

摘自xyz32768 的博客

Code：

#include<cstdio>
#include<deque>
#include<cstring>
#include<algorithm>
#define maxn 205
#define maxm 50005
#define LL long long
using namespace std;
const int inf = 0x3f3f3f3f;
int n,m,S,T;
int fir[maxn],cur[maxn],nxt[maxm],to[maxm],c[maxm],tot=1;LL w[maxm];
inline void line(int x,int y,int z,LL v){
    nxt[++tot]=fir[x],fir[x]=tot,to[tot]=y,c[tot]=z,w[tot]=v;
    nxt[++tot]=fir[y],fir[y]=tot,to[tot]=x,c[tot]=0,w[tot]=-v;
}
namespace Costflow{
    LL dis[maxn],sum;
    bool vis[maxn],inq[maxn];
    deque<int>q;
    bool SPFA(){
        memset(dis,-0x3f,(T+1)<<3);
        dis[T]=0,q.push_back(T);
        while(!q.empty()){
            int u=q.front();q.pop_front(),inq[u]=0;
            for(int i=fir[u],v;i;i=nxt[i]) if(c[i^1]&&dis[v=to[i]]<dis[u]+w[i^1]){
                dis[v]=dis[u]+w[i^1];
                if(!inq[v]) {inq[v]=1;if(!q.empty()&&dis[v]>dis[q.front()]) q.push_front(v);else q.push_back(v);}
            }
        }
        return sum+dis[S]>=0;
    }
    int aug(int u,int augco){
        if(u==T) return augco;
        vis[u]=1;
        int need=augco,delta;
        for(int &i=cur[u];i;i=nxt[i]) if(c[i]&&!vis[to[i]]&&dis[u]==dis[to[i]]+w[i]){
            delta=aug(to[i],min(need,c[i])),c[i]-=delta,c[i^1]+=delta;
            if(!(need-=delta)) break;
        }
        return vis[u]=0,augco-need;
    }
    int solve(){
        int flow=0,ans=0;
        while(SPFA()){
            memcpy(cur,fir,(T+1)<<2),flow=aug(S,inf);
            if(sum+dis[S]*flow>=0) ans+=flow,sum+=dis[S]*flow;
            else {ans+=sum/(-dis[S]);break;}
        }
        return ans;
    }
}
int cnt[maxn],a[maxn],b[maxn],cc[maxn];
int Cut(int x){
    int ret=0;
    for(int i=2;i*i<=x;i++) while(x%i==0) x/=i,ret++;
    return ret+(x>1);
}
int main()
{
    scanf("%d",&n),S=0,T=n+1;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),cnt[i]=Cut(a[i]);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]),cnt[i]&1?(line(S,i,b[i],0),0):(line(i,T,b[i],0),0);
    for(int i=1;i<=n;i++) scanf("%d",&cc[i]);
    for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) if(a[j]%a[i]==0&&cnt[j]==cnt[i]+1){
        if(cnt[i]&1) line(i,j,inf,1ll*cc[i]*cc[j]);
        else line(j,i,inf,1ll*cc[i]*cc[j]);
    }
    printf("%d\n",Costflow::solve());
}