[BZOJ3550][ONTAK2010]Vacation

原题地址

MS又是一道费用流神题，然而单纯形直接搞…

不用对偶了简直感动(然而还是没有机会试验我的init函数)…

AC code:

#include <cstdio>
#include <cmath>
typedef double llf;
const int N=2010;
const int INF=1<<29;
const llf eps=1e-7;
int n,m,k;
llf v;
llf b[N],c[N];
llf A[N][N];

void pivot(int l,int e){
    b[l]/=A[l][e];
    for(int i=1;i<=m;i++) if(i!=e) A[l][i]/=A[l][e];
    A[l][e]=1/A[l][e];
    for(int i=1;i<=n;i++){
        if(i==l||fabs(A[i][e])<eps) continue;
        b[i]-=A[i][e]*b[l];
        for(int j=1;j<=m;j++) if(j!=e) A[i][j]-=A[i][e]*A[l][j];
        A[i][e]*=-A[l][e];
    }
    v+=c[e]*b[l];
    for(int i=1;i<=m;i++) if(i!=e) c[i]-=c[e]*A[l][i];
    c[e]*=-A[l][e];
}

void opt(){
    while(1){
        int e=INF,l=INF;
        llf tmp=INF;
        for(int i=1;i<=m;i++){
            if(c[i]>eps){
                e=i;
                break;
            }
        }
        if(e==INF) return ;
        for(int i=1;i<=n;i++){
            if(A[i][e]>eps&&b[i]/A[i][e]<tmp){
                l=i;
                tmp=b[i]/A[i][e];
            }
        }
        pivot(l,e);
    }
}

void simplex(){
    opt();
    printf("%d\n",(int)(v+0.5));
}

int main(){
    scanf("%d%d",&m,&k);
    m*=3;
    for(int i=1;i<=m;i++) scanf("%lf",&c[i]);
    for(int i=1;i<=m;i++) {A[i][i]=1;b[i]=1;}
    for(int i=1;i<=m/3*2+1;i++){
        for(int j=0;j<m/3;j++) A[m+i][j+i]=1;
        b[m+i]=k;
    }
    n=m+m/3*2+1;
    simplex();

    return 0;
}