BZOJ4514：数字配对（费用流+数学）

《Clannad》告诉我，即使有的事再努力也没有用，也要去拼命争取。
无论结果如何，我也不会后悔那次在坡道上与你的相遇。
最后半年的oi之路，就从这个白色相簿的季节开始吧。

题面
题意：n个数，若a[i]/a[j]=p，p为质数，则a[i]和a[j]可以配对，产生c[i]*c[j]的价值，a[i]可以配对b[i]次，问在价值＞0的前提下，最多配对多少次。

不难，题目条件很符合费用流的格式。
对于A和B能配对，B和C能配对，假设A=B*p，B=C*q，则A和C不能配对。
无论怎么假设，都发现A和C不能配对。

于是我们按a的质因数个数的奇偶把数分成两类，同类的都不能配对。
然后建二分图跑费用流就可以了。

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int T=401,nn=1000,N=1010,M=1001000,oo=1e9;
const LL ooo=1e18;

int n,a[N],b[N];
LL c[N];
int f[N];
int head[N],to[M],nex[M],cap[M],cnt=2;
int q[N],flow[N],id[N],pre[N];
bool in[N];
LL val[M],d[N];

void add(int u,int v,int ca,LL va)
{
    to[cnt]=v;
    cap[cnt]=ca;
    val[cnt]=va;
    nex[cnt]=head[u];
    head[u]=cnt++;
    to[cnt]=u;
    cap[cnt]=0;
    val[cnt]=-va;
    nex[cnt]=head[v];
    head[v]=cnt++;
}

bool spfa()
{
    for(int i=1;i<=T;i++)
    d[i]=ooo;
    mmst(in,0);
    d[0]=0;
    flow[0]=oo;
    int hh=1,tt=1;
    q[1]=0;
    while(hh<=tt)
    {
        int hy=q[hh%nn];
        hh++;
        for(int h=head[hy];h;h=nex[h])
        if(cap[h]&&d[to[h]]>d[hy]+val[h])
        {
            d[to[h]]=d[hy]+val[h];
            id[to[h]]=h;
            pre[to[h]]=h;
            flow[to[h]]=min(flow[hy],cap[h]);
            if(!in[to[h]])
            {
                in[to[h]]=1;
                tt++;
                q[tt%T]=to[h];
                if(d[q[tt%nn]]<d[q[hh%nn]])
                swap(d[q[tt%nn]],d[q[hh%nn]]);
            }
        }
    }
    return d[T]<oo;
}

bool prime(int x)
{
    for(int i=2;i*i<=x;i++)
    if(x%i==0)
    return 0;
    return 1;
}

int work(int x)
{
    int res=0,hy=sqrt(x);
    for(int i=2;i<=hy;i++)
    while(x%i==0)
    {
        x/=i;
        res^=1;
    }
    if(x!=1)
    res^=1;
    return res;
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);

    for(int i=1;i<=n;i++)
    scanf("%d",&b[i]);

    for(int i=1;i<=n;i++)
    scanf("%lld",&c[i]);

    for(int i=1;i<=n;i++)
    f[i]=work(a[i]);

    for(int i=1;i<=n;i++)
    if(!f[i])
    add(0,i,b[i],0);
    else
    add(i,T,b[i],0);

    for(int i=1;i<=n;i++)
    if(!f[i])
    for(int j=1;j<=n;j++)
    if(f[j])
    {
        int x=min(a[i],a[j]),y=max(a[i],a[j]);
        if(y%x==0&&prime(y/x))
        add(i,j,oo,-c[i]*c[j]);
    }

    int ans=0;
    LL sum=0;
    while(spfa())
    {
        LL now=d[T]*flow[T];
        if(now<=-sum)
        ans+=flow[T],sum+=now;
        else
        {
            LL tu=(-sum)/d[T];
            ans+=tu;
            break;
        }
        for(int i=T;i;i=pre[i])
        {
            cap[id[i]]-=flow[T];
            cap[id[i]^1]+=flow[T];
        }
    }
    cout<<ans<<endl;

    return 0;
}

你够资格吗，也许不够