bzoj4514: [Sdoi2016]數字配對(費用流)

Hanks_o發表於2018-03-05

題目傳送門

解法:
哇跟網上做法完全不一樣誒。
我本來是這樣想的。
分成兩排點,都是1~n。
能連的話第一排連第二排。
這樣的話出現了一個問題。如果第二個數字只能用一次。
第一個數字去跟第二個數字匹配。第二個數字去跟第三個數字匹配。
這種情況我無法限制。

然後。我就想到一個比較討巧的方法。
還是兩排。
能連的話第一排連第二排。然後反過來的也要連。
這樣的話就不會出現上一種情況了。
如果第一個數字去跟第二個數字匹配。表示這個費用是比較大的。
那麼第二個數字去跟第一個數字匹配其實是一樣的。所以會再加多一次答案。
最後答案除以2就行。。

怎麼判斷兩個數是否滿足條件匹配。
先把每個數質因數次數的和算出來。
如果能整除然後次數和相差1那麼肯定相差一個質數沒乘。然後就做出來了。。

翻了下跟網上的做法完全不一樣嘿嘿嘿很開心啊。。

程式碼實現:

#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node {int x,y,c;ll d,next,other;}a[110000];int len,last[1100];
void ins(int x,int y,int c,ll d) {
    int k1,k2;
    len++;k1=len;a[len].x=x;a[len].y=y;a[len].c=c;a[len].d=d;
    a[len].next=last[x];last[x]=len;
    len++;k2=len;a[len].x=y;a[len].y=x;a[len].c=0;a[len].d=-d;
    a[len].next=last[y];last[y]=len;
    a[k1].other=k2;a[k2].other=k1;
}
int list[1100],head,tail,fbian[1100],st,ed;bool v[1100];ll d[1100];
bool spfa() {
    head=1;tail=2;list[1]=st;memset(v,false,sizeof(v));v[st]=true;
    for(int i=1;i<=ed;i++)d[i]=-99999999999;d[st]=0;
    while(head!=tail) {
        int x=list[head];
        for(int k=last[x];k;k=a[k].next) {
            int y=a[k].y;
            if(a[k].c>0&&d[y]<d[x]+a[k].d) {
                d[y]=d[x]+a[k].d;fbian[y]=k;
                if(v[y]==false) {v[y]=true;list[tail++]=y;if(tail==ed+1)tail=1;}
            }
        }head++;v[x]=false;if(head==ed+1)head=1;
    }return d[ed]==-99999999999?false:true;
}
int ans=0;ll sum=0;bool bk=false;
void wh() {
    int x=ed,flow=999999999;
    while(x!=st) {int k=fbian[x];flow=min(flow,a[k].c);x=a[k].x;}
    if(sum+flow*d[ed]>=0) {sum+=flow*d[ed];ans+=flow;}
    else {ans+=sum/(-d[ed]);bk=true;}
    x=ed;while(x!=st) {int k=fbian[x];a[k].c-=flow;a[a[k].other].c+=flow;x=a[k].x;}
    return;
}
int n;
int A[210],B[210];ll C[210];
int f[210];
int main() {
    scanf("%d",&n);st=2*n+1;ed=st+1;
    for(int i=1;i<=n;i++) {
        scanf("%d",&A[i]);
        int t=int(sqrt(double(A[i]+1)));int x=A[i];
        for(int j=2;j<=t;j++) {
            while(x%j==0) {x/=j;f[i]++;}
        }if(x!=1)f[i]++;
    }len=0;memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++) {scanf("%d",&B[i]);ins(st,i,B[i],0);ins(i+n,ed,B[i],0);}
    for(int i=1;i<=n;i++) scanf("%lld",&C[i]);
    for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)if(i!=j) {
        if(A[i]%A[j]==0&&f[i]-f[j]==1) {
            ins(i,j+n,999999999,C[i]*C[j]);
            ins(j,i+n,999999999,C[i]*C[j]);
        }
    }bk=false;
    while(spfa()==true) {wh();if(bk==true)break;}
    printf("%d\n",ans/2);
    return 0;
}

相關文章