km算法模板

输入数据包含多组测试用例，每组数据的第一行输入n,表示男生的数量，接下来有n行,每行n个数表示第i个女生对第j个男生的好感度(n<=300)

        我们用fill函数来给slack赋极大值

#include<iostream>//slack:匹配
#include<algorithm>//match:松弛
#include<cstring>//gap:差距
#include<cstdio>
#define inf 0x3f3f3f3f
using namespace std;

int n;
int love[301][301];   ///记录每个女生对男生的好感度
int vis_girl[301],vis_boy[301];///记录是否匹配过女生,男生
int ex_girl[301],ex_boy[301];///记录女生,男生的期望值
int match[301],slack[301];///记录每个男生匹配的女生,如果没有就-1,记录每个男生如果能被女生倾心最少还需要多少期望值

int dfs(int x)
{
    vis_girl[x]=1;
    for(int i=0;i<n;i++)
    {
        if(vis_boy[i]) continue;///每一轮匹配,男生只尝试一次
        int gap=ex_girl[x]+ex_boy[i]-love[x][i];
        if(!gap){///如果符合要求
            vis_boy[i]=1;
            if(match[i]==-1||dfs(match[i])){///找到可以匹配的男生||该女生的男朋友可以找其他人
                match[i]=x;
                return 1;
            }
        }
        else slack[i]=min(slack[i],gap);/// slack 可以理解为该男生要得到女生的倾心 还需多少期望值 取最小值 备胎的样子【捂脸
    }
    return 0;
}

void km()
{
    memset(match,-1,sizeof(match));///所以男生都没女朋友
    memset(ex_boy,0,sizeof(ex_boy));///每个男生的期望值都为零
    for(int i=0;i<n;i++)///女生的期望值为她对男生的最大好感度
    {
        ex_girl[i]=love[i][0];
        for(int j=1;j<n;j++)
            ex_girl[i]=max(ex_girl[i],love[i][j]);
    }

    for(int i=0;i<n;i++)///尝试为每个女生找男朋友
    {
        fill(slack,slack+n,inf);///用fill给slack赋值
        while(1){ ///寻找方法:找不到就降低期望值,直到找到为止
            memset(vis_boy,0,sizeof(vis_boy));  ///重置访问记录
            memset(vis_girl,0,sizeof(vis_girl));
            if(dfs(i)) break;    ///找到归宿就结束
            int d=inf;
            for(int j=0;j<n;j++) ///寻找最低需要降低的期望值
                if(!vis_boy[j]) d=min(d,slack[j]);
            for(int j=0;j<n;j++){
                if(vis_girl[j]) ex_girl[j]-=d;///访问过的所以女生降低期望值
                if(vis_boy[j]) ex_boy[j]+=d;///访问过的所以男生降低期望值
                else slack[j]-=d;         
            }
        }
    }
    int res=0;
    for(int i=0;i<n;i++)
        res+=love[match[i]][i];
    printf("%d\n",res);
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&love[i][j]);
        km();
    }
    return 0;
}

#include<iostream>//slack:匹配
#include<algorithm>//match:松弛
#include<cstring>//gap:差距
#include<cstdio>
#define inf 0x3f3f3f3f
using namespace std;

int n;
int love[301][301];
int vis_girl[301],vis_boy[301];
int ex_girl[301],ex_boy[301];
int match[301],slack[301];

int dfs(int x)
{
    vis_girl[x]=1;
    for(int i=0;i<n;i++)
    {
        if(vis_boy[i]) continue;
        int gap=ex_girl[x]+ex_boy[i]-love[x][i];
        if(!gap){
            vis_boy[i]=1;
            if(match[i]==-1||dfs(match[i])){
                match[i]=x;
                return 1;
            }
        }
        else slack[i]=min(slack[i],gap);
    }
    return 0;
}

void km()
{
    memset(match,-1,sizeof(match));
    memset(ex_boy,0,sizeof(ex_boy));
    for(int i=0;i<n;i++)
    {
        ex_girl[i]=love[i][0];
        for(int j=1;j<n;j++)
            ex_girl[i]=max(ex_girl[i],love[i][j]);
    }

    for(int i=0;i<n;i++)
    {
        fill(slack,slack+n,inf);
        while(1){
            memset(vis_boy,0,sizeof(vis_boy));
            memset(vis_girl,0,sizeof(vis_girl));
            if(dfs(i)) break;
            int d=inf;
            for(int j=0;j<n;j++)
                if(!vis_boy[j]) d=min(d,slack[j]);
            for(int j=0;j<n;j++){
                if(vis_girl[j]) ex_girl[j]-=d;
                if(vis_boy[j]) ex_boy[j]+=d;
                else slack[j]-=d;
            }
        }
    }
    int res=0;
    for(int i=0;i<n;i++)
        res+=love[match[i]][i];
    printf("%d\n",res);
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&love[i][j]);
        km();
    }
    return 0;
}

下面这道km算法题需要自己构图: hdu2255(Chocolate,km算法+构图) 

转载于:https://www.cnblogs.com/wangtao971115/p/10358355.html