p4068 [SDOI2016]数字配对

传送门

分析

我们考虑对所有a[i]质因数分解，然后记cnt[i]为a[i]是由几个质数相乘得到的

然后我们建一个二分图，左面为所有cnt[i]为奇数的点，右面是为偶数的点

我们从源点向左面点连容量b[i]花费0的边，从右面点向汇点连容量b[i]花费0的边

然后两排点之间符合条件的点对连容量inf花费c[i]*c[j]的边

然后跑总花费不小于0的最大费用最大流即可

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define int long long
const int inf = 1e15+;
int n,a[],b[],C[],cnt[];
int x[],y[],l1,l2,s,t,Ans,now,sum;
int head[],w[],c[],nxt[],to[];
int fm[],ano[],S;
inline void add(int x,int y,int z,int cost){
    nxt[++S]=head[x];head[x]=S;to[S]=y;w[S]=z;c[S]=cost;fm[S]=x;
    nxt[++S]=head[y];head[y]=S;to[S]=x;w[S]=;c[S]=-cost;fm[S]=y;
    ano[S]=S-;ano[S-]=S;
}
inline int work(int x){
    int i,j,k=;
    for(i=;i<=sqrt(x);i++)
      while(x%i==)x/=i,k++;
    if(x>)k++;
    return k;
}
queue<int>q;
int iq[],nf[],la[],d[];
inline void go(){
    int i,j,k;
    la[s]=;
    while(){
      for(i=;i<=t;i++)d[i]=-inf;
      q.push(s);
      d[s]=,nf[s]=inf,iq[s]=;
      while(!q.empty()){
          int u=q.front();
          q.pop(),iq[u]=;
          for(i=head[u];i;i=nxt[i])
            if(w[i]&&d[to[i]]<d[u]+c[i]){
              d[to[i]]=d[u]+c[i];
              la[to[i]]=i;
              nf[to[i]]=min(w[i],nf[u]);
              if(!iq[to[i]]){
                q.push(to[i]);
                iq[to[i]]=;
              }
            }
      }
      int be=now,ans=Ans,i=la[t];
      if(!nf[t]||d[t]==-inf)return;
      while(i){
          w[i]-=nf[t];
          w[ano[i]]+=nf[t];
          now+=nf[t]*c[i];
          i=la[fm[i]];
      }
      Ans+=nf[t];
      if(now<){
          Ans=ans+be/abs(d[t]);
          return;
      }
    }
}
signed main(){
    int i,j,k;
    scanf("%lld",&n);
    s=n+,t=n+;
    for(i=;i<=n;i++)scanf("%lld",&a[i]),cnt[i]=work(a[i]);
    for(i=;i<=n;i++)scanf("%lld",&b[i]);
    for(i=;i<=n;i++)scanf("%lld",&C[i]);
    for(i=;i<=n;i++)if(cnt[i]&)
      for(j=;j<=n;j++)
        if((cnt[i]==cnt[j]+&&a[i]%a[j]==)
          ||(cnt[j]==cnt[i]+&&a[j]%a[i]==))add(i,j,inf,C[i]*C[j]);
    for(i=;i<=n;i++)
      if(cnt[i]&)add(s,i,b[i],);
        else add(i,t,b[i],);
    go();
    printf("%lld\n",Ans);
    return ;
}