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

来源：互联网发布：淘宝换货运费险怎么退编辑：程序博客网时间：2024/06/05 07:13

这道题只要看出是个二分图就可以直接费用流搞一搞了
~~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 1e9using 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;}

0 0