bzoj 4514: [Sdoi2016]数字配对 费用流

题意

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

分析

设f(x)表示x的标准分解式的指数之和，那么i和j匹配则必满足a[i]%a[j]==0且f(a[i])==f(a[j])+1
那么我们就可以把i按照f(a[i])的奇偶性来黑白染色。
连边之后二分答案然后判断最大费用最大流是否不小于0即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define N 205
#define LL long long
#define inf 0x7fffffff
using namespace std;

int n,a[N],b[N],c[N],cnt,last[N],s,t,S,pre[N],vis[N],col[N],maxf,sum,prime[N*N],not_prime[N*N],low[N*N];
LL dis[N],ans;
struct edge{int from,to,next,c;LL w;}e[N*N*5];
queue <int> q;

void get_prime(int n)
{
    for (int i=2;i<=n;i++)
    {
        if (!not_prime[i]) prime[++sum]=i,low[i]=sum;
        for (int j=1;j<=sum&&i*prime[j]<=n;j++)
        {
            not_prime[i*prime[j]]=1;low[i*prime[j]]=prime[j];
            if (i%prime[j]==0) break;
        }
    }
}

void addedge(int u,int v,int c,LL w)
{
    e[++cnt].from=u;e[cnt].to=v;e[cnt].c=c;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].from=v;e[cnt].to=u;e[cnt].c=0;e[cnt].w=-w;e[cnt].next=last[v];last[v]=cnt;
}

bool spfa()
{
    for (int i=0;i<=S;i++) dis[i]=-1e15;
    q.push(S);vis[S]=1;dis[S]=0;
    while (!q.empty())
    {
        int u=q.front();
        q.pop();
        for (int i=last[u];i;i=e[i].next)
            if (e[i].c&&dis[u]+e[i].w>dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+e[i].w;
                pre[e[i].to]=i;
                if (!vis[e[i].to])
                {
                    vis[e[i].to]=1;
                    q.push(e[i].to);
                }
            }
        vis[u]=0;
    }
    if (dis[t]==-1e15) return 0;
    else return 1;
}

void mcf()
{
    int mn=inf,x=t;
    while (pre[x])
    {
        mn=min(mn,e[pre[x]].c);
        x=e[pre[x]].from;
    }
    ans+=(LL)mn*dis[t];
    maxf+=mn;
    x=t;
    while (pre[x])
    {
        e[pre[x]].c-=mn;
        e[pre[x]^1].c+=mn;
        x=e[pre[x]].from;
    }
}

bool check(int mid)
{
    e[cnt^1].c=mid;e[cnt].c=0;
    for (int i=2;i<=cnt;i+=2) e[i].c+=e[i^1].c,e[i^1].c=0;
    ans=0;maxf=0;
    while (spfa()) mcf();
    if (ans>=0&&maxf==mid) return 1;
    else return 0;
}

int main()
{
    get_prime(40000);
    scanf("%d",&n);
    int tot=0;
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<=n;i++) scanf("%d",&b[i]),tot+=b[i];
    for (int i=1;i<=n;i++) scanf("%d",&c[i]);
    for (int i=1;i<=n;i++)
    {
        int x=a[i];
        for (int j=1;j<=sum;j++)
            if (x%prime[j]==0)
            {
                while (x%prime[j]==0)
                {
                    col[i]++;x/=prime[j];
                }
            }
    }
    cnt=1;
    for (int i=1;i<=n;i++)
        if (col[i]%2==0)
            for (int j=1;j<=n;j++)
                if (col[j]%2==1)
                {
                    int x=i,y=j;
                    if (a[x]>a[y]) swap(x,y);
                    if (a[y]%a[x]==0&&col[x]+1==col[y]) addedge(i,j,inf,(LL)c[i]*c[j]);
                }
    s=0;t=n+1;S=n+2;
    for (int i=1;i<=n;i++)
        if (col[i]%2==0) addedge(s,i,b[i],0);
        else addedge(i,t,b[i],0);
    addedge(S,s,0,0);
    int l=1,r=tot;
    while (l<=r)
    {
        int mid=(l+r)/2;
        if (check(mid)) l=mid+1;
        else r=mid-1;
    }
    printf("%d",l-1);
    return 0;
}