奔小康赚大钱

竟然卡最小费用最大流。
因为是稠密图的原因吗?
那我们呢就用KM算法吧。
这一用不要紧,我竟然发现我的KM模板竟然也被卡时间了。。。。。。
赶紧更新自己的模板。

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef pair<int,int> pii;
const int max_n = 1000;
const int max_m = 1200;
const int inf = 1e9;
int n,nx,ny;
int linker[max_n],lx[max_n],ly[max_n],slack[max_n]; 
int visx[max_n],visy[max_n],w[max_n][max_n];

int DFS(int x){
    visx[x]=1;
    for(int y=1;y<=ny;y++){
        if(visy[y])
            continue;
        int tmp=lx[x]+ly[y]-w[x][y];
        if(tmp==0){
            visy[y]=1;
            if(linker[y]==-1 || DFS(linker[y])){
                linker[y]=x;
                return 1;
            }
        }else if(slack[y]>tmp){
            slack[y]=tmp;
        }
    }
    return 0;
}

int KM(){
    int i,j;
    memset(linker,-1,sizeof(linker));
    memset(ly,0,sizeof(ly));
    for(i=1;i<=nx;i++) 
        for(j=1,lx[i]=-inf;j<=ny;j++)
            if(w[i][j]>lx[i])
                lx[i]=w[i][j];
    for(int x=1;x<=nx;x++){
        for(i=1;i<=ny;i++)
            slack[i]=inf;
        while(1){
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(DFS(x)) 
                break;  

            int d=inf;
            for(i=1;i<=ny;i++)
                if(!visy[i] && d>slack[i])
                    d=slack[i];
            for(i=1;i<=nx;i++)
                if(visx[i])
                    lx[i]-=d;
            for(i=1;i<=ny;i++)
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
        }
    }
    int res=0;
    for(i=1;i<=ny;i++)
        if(linker[i]!=-1)
            res+=w[linker[i]][i];
    return res;
}
int main(){
    while(~scanf("%d",&n)){
        nx=ny=n;
        for (int i=1;i<=n;++i){
            for (int j=1;j<=n;++j){
                scanf("%d",&w[i][j]);
            }
        }
        printf("%d\n",KM());
    }
}

题单:https://vjudge.net/article/371

全部评论

相关推荐

威猛的小饼干正在背八股:挂到根本不想整理
点赞 评论 收藏
分享
2 收藏 评论
分享
牛客网
牛客企业服务