4514: [Sdoi2016]数字配对

题目链接

题目大意：$n$种数字，第 $i$ 种数字是 $a_i$，有 $b_i$ 个，权值是 $c_i$
若两个数字 $a_i、a_j$ 满足，$a_i$是 $a_j$ 的倍数，且 $a_i/a_j$ 是一个质数，
那么这两个数字可以配对，并获得 $c_i×c_j$ 的价值。一个数字只能参与一次配对
求价值和不小于0的前提下的最多配对次数

题解：按照质因子个数的奇偶性建立二分图，价值的要求需要手动特判一下

我的收获：~~套路

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <set>
using namespace std;

const int M=205*2;
const long long INF=1e15;

int n;
int a[M],b[M],c[M],mo[M];
long long flow,mincost;
int t,head[M],last[M];
int st,ed;
long long d[M],pre[M];
int p[100005];
bool flag,vis[M],isp[100005];

queue<int> q;
set<int> mp;

struct edge{int fro,to,nex,c;long long val;}e[M*M];

void add(int i,int j,int ca,long long co){e[t].fro=i,e[t].to=j,e[t].nex=head[i],e[t].c=ca,e[t].val=co;head[i]=t++;}
void insert(int i,int j,int w,long long z){add(i,j,w,z),add(j,i,0,-z);}

void ela(){
    for(int i=2;i<=32000;i++){
        if(!isp[i]) p[++p[0]]=i,mp.insert(i);
        for(int j=1;j<=p[0]&&i*p[j]<=32000;j++){
            isp[i*p[j]]=1;
            if(i%p[j]==0) break;
        }
    }
}

void fj(int id)
{
    int x=a[id];
    for(int i=1;i<=p[0];i++)
        while(x%p[i]==0) mo[id]++,x/=p[i];
    if(x!=1) mo[id]++,mp.insert(x);
}

bool spfa()
{
    for(int i=0;i<=ed;i++) vis[i]=0,d[i]=INF;
    d[st]=0;q.push(st);
    while(!q.empty())
    {
        int u=q.front();q.pop();vis[u]=false;
        for(int i=head[u];i!=-1;i=e[i].nex){
            int v=e[i].to;
            if(e[i].c&&d[v]>d[u]+e[i].val){
                d[v]=d[u]+e[i].val;
                pre[v]=i;
                if(!vis[v]){
                    vis[v]=true;
                    q.push(v);
                }
            }
        }
    }
    return d[ed]!=INF;
}

void mcfx()
{
    int mx=0x3f3f3f3f;long long valnow=0;
    for(int u=ed;u!=st;u=e[pre[u]].fro)
        mx=min(mx,e[pre[u]].c);
    for(int u=ed;u!=st;u=e[pre[u]].fro){
        e[pre[u]].c-=mx;e[pre[u]^1].c+=mx;
        valnow+=e[pre[u]].val*mx;
    }
    if(mincost+valnow<=0) mincost+=valnow,flow+=mx;
    else{flag=1;flow+=-mincost/(d[ed]);return ;}
}

void work()
{
    while(spfa()&&flag==0) mcfx();
    printf("%d\n",flow);
}

void build()
{
    for(int i=1;i<=n;i++) fj(i);
    for(int i=1;i<=n;i++) 
    if(mo[i]&1) insert(st,i,b[i],0);
    else insert(i,ed,b[i],0);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
        if(a[i]%a[j]==0&&mp.find(a[i]/a[j])!=mp.end())
        {
            if(mo[i]&1) insert(i,j,0x3f3f3f3f,-1ll*c[i]*c[j]);
            else insert(j,i,0x3f3f3f3f,-1ll*c[i]*c[j]);
        }
    }
}

void init()
{
    memset(head,-1,sizeof(head));
    scanf("%d",&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("%d",&c[i]);
    st=0,ed=n+1;
    ela();build();
}

int main()
{
    init();
    work();
    return 0;
}