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

这道题只要看出是个二分图就可以直接费用流搞一搞了
SDOI的考场上我是看出二分图的,为什么还是打的暴力。。我sb

#include<algorithm>
#include<iostream>
#include<complex>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#include<ctime>
#include<set>
#include<map>
#define N 222222
#define inf 1e9
using namespace std;
int sc()
{
    int i=0,f=1;char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
int pr[10]={2,3,5,7,11,13,17,19,23,29};
bool prime(int x)
{
    for(int i=0;i<10;i++)
    {
        if(x==pr[i])return 1;
        long long ans=1,p=pr[i],y=x-1;
        for(;y;y>>=1,p=p*p%x)
            if(y&1)ans=ans*p%x;
        if(ans!=1)return 0;
    }
    return 1;
}
struct W{int a,b,c;}a[N];
long long v[N],dis[N],ans,sum;
int q[N],from[N],wh[N],mark[N],col[N];
int Head[N],Nxt[N],Lst[N],Tot;
int head[N],nxt[N],lst[N],c[N],tot=1;
int n,m,S,T;
void insert(int x,int y,int a,long long b)
{
    lst[++tot]=y;nxt[tot]=head[x];head[x]=tot;c[tot]=a;v[tot]=b;
    lst[++tot]=x;nxt[tot]=head[y];head[y]=tot;c[tot]=0;v[tot]=-b;
}
void INSERT(int x,int y)
{
    Lst[++Tot]=y;Nxt[Tot]=Head[x];Head[x]=Tot;
    Lst[++Tot]=x;Nxt[Tot]=Head[y];Head[y]=Tot;
}
bool cmp(W a,W b)
{
    return a.a<b.a;
}
void dfs(int x)
{
    for(int i=Head[x];i;i=Nxt[i])
        if(!col[Lst[i]])col[Lst[i]]=col[x]^1,dfs(Lst[i]);
}
bool spfa()
{
    for(int i=1;i<=T;i++)dis[i]=-1ll<<59;
    dis[q[1]=S]=0;
    for(int l=1,r=2;l!=r;l=(l+1)%N)
    {
        int x=q[l];mark[x]=0;
        for(int i=head[x];i;i=nxt[i])
            if(c[i]&&dis[lst[i]]<dis[x]+v[i])
            {
                dis[lst[i]]=dis[x]+v[i];
                from[lst[i]]=x;
                wh[lst[i]]=i;
                if(!mark[lst[i]])
                     mark[q[r]=lst[i]]=1,r=(r+1)%N;
            }
    }
    return dis[T]!=-1ll<<59;
}
int main()
{
    n=sc();S=n+1,T=n+2;
    for(int i=1;i<=n;i++)a[i].a=sc();
    for(int i=1;i<=n;i++)a[i].b=sc();
    for(int i=1;i<=n;i++)a[i].c=sc();
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
            if(a[j].a%a[i].a==0&&prime(a[j].a/a[i].a))
                 INSERT(i,j);
    for(int i=1;i<=n;i++)
        if(!col[i])col[i]=2,dfs(i);
    for(int i=1;i<=n;i++)
        if(col[i]&1)insert(i,T,a[i].b,0);
        else
        {
            insert(S,i,a[i].b,0);
            for(int j=Head[i];j;j=Nxt[j])
                insert(i,Lst[j],inf,(long long)a[i].c*a[Lst[j]].c);
        }
    while(spfa())
    {
        int mn=1e9;
        for(int i=T;i!=S;i=from[i])mn=min(mn,c[wh[i]]);
        for(int i=T;i!=S;i=from[i])c[wh[i]]-=mn,c[wh[i]^1]+=mn;
        if(sum+dis[T]*mn<0)
        {
            ans-=sum/dis[T];
            break;
        }
        else ans+=mn,sum+=dis[T]*mn;
    }
    cout<<ans;
    return 0;
}