[hdu2255]奔小康赚大钱(二分图最优匹配、KM算法)

最新推荐文章于 2024-10-18 11:16:53 发布

weixin_30627341

最新推荐文章于 2024-10-18 11:16:53 发布

阅读量73

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/elpsycongroo/p/8975747.html

本文介绍了一种解决二分图最优匹配问题的方法，即KM算法。该算法首先确保匹配的数量达到最大，其次使匹配边的权值之和最大。通过详细的C++代码示例展示了如何实现这一算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：求二分图的最优匹配（首先数目最大，其次权值最大）。

解题关键：KM算法

复杂度：$O(n^3)$

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iostream>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=310;
const int inf=0x3f3f3f3f;
int nx,ny;
int g[N][N];
int match[N],lx[N],ly[N];//y中各点匹配状态，x,y中的顶点标号
int slack[N];
bool visx[N],visy[N];

bool hungry(int x){
    visx[x]=1;
    for(int y=1; y<=ny; y++){
        if(visy[y])continue;
        int tmp=lx[x]+ly[y]-g[x][y];
        if(tmp==0){
            visy[y]=true;
            if(match[y]==-1 ||hungry(match[y])){
                match[y]=x;
                return 1;
            }
        }
        else if(slack[y]>tmp) slack[y]=tmp;
    }
    return 0;
}

int KM(){
    memset(match, -1, sizeof match);
    memset(ly, 0, sizeof ly);
    for(int i=1;i<=nx;i++){
        lx[i]=-inf;
        for(int j=1;j<=ny;j++) lx[i]=max(lx[i],g[i][j]);
    }
    for(int x=1;x<=nx;x++){
        for(int i=1;i<=ny;i++) slack[i]=inf;
        while(1){
            memset(visx,0,sizeof visx);
            memset(visy,0,sizeof visy);
            if(hungry(x))break;
            int d=inf;
            for(int i=1;i<=ny;i++)if(!visy[i]&&d>slack[i]) d=slack[i];
            for(int i=1;i<=nx;i++)if(visx[i])lx[i]-=d;
            for(int i=1;i<=ny;i++){
                if(visy[i])ly[i]+=d;
                else slack[i]-=d;
            }
        }
    }
    int res=0;
    for(int i=1;i<=ny;i++)
        if(match[i]!=-1) res+=g[match[i]][i];
    return res;
}

int main(){
    int n;
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&g[i][j]);
        nx=ny=n;
        printf("%d\n",KM());
    }
    return 0;
}