kuangbin专题十 HDU2255 KM算法模板

题解:
KM算法的模板题,主要学习的两个博客为:
第一个是看他解说KM算法:
https://www.cnblogs.com/Lanly/p/6291214.html
第二个是看他的代码和模拟:
https://www.cnblogs.com/wenruo/p/5264235.html

不多说了,我感觉我来说还不如你们看上面两个更好。

/*KM算法
 O(nx*nx*ny)
 求最大权匹配
 若求最小权匹配,可将权值取相反数,结果再取相反数。
*/
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f 
const int MAXN=310;
int link[MAXN][MAXN];
int lx[MAXN],ly[MAXN];//x,y中的顶标
int slack[MAXN];
bool visx[MAXN],visy[MAXN];
int cx[MAXN],cy[MAXN];
int nx,ny,n;
bool dfs(int u)
{
    visx[u]=true;
    for(int v=1;v<=ny;v++)
    {
        if(visy[v])
        continue;
        int tmp=lx[u]+ly[v]-link[u][v];
        if(tmp==0)
        {
            visy[v]=true;
            if(cy[v]==-1||dfs(cy[v]))
            {
                cx[u]=v;
                cy[v]=u;
                return true;
            }
        }
        else
        slack[v]=min(slack[v],tmp);
    }
    return false;
}
int KM()
{
    memset(cy,-1,sizeof(cy));
    memset(cx,-1,sizeof(cx));
    memset(ly,0,sizeof(ly));
    for(int i=1;i<=nx;i++)
    {
        lx[i]=-INF;
        for(int j=1;j<=ny;j++)
        if(link[i][j]>lx[i])
        lx[i]=link[i][j];
    }
    for(int i=1;i<=nx;i++)
    {
    //  memset(slack,INF,sizeof(slack));
        fill(slack+1,slack+ny+1,INF);   //黑科技,fill可以赋任何值,而memst只能赋0,-1 
    //  for(int j=1;j<=ny;j++)
    //  slack[j]=INF;
        while(1)
        {
            memset(visx,false,sizeof(visx));
            memset(visy,false,sizeof(visy));
            if(dfs(i)) break;
            int d=INF;
            for(int j=1;j<=ny;j++)
            if(!visy[j]) 
            d=min(d,slack[j]);
            for(int j=1;j<=nx;j++)
            if(visx[j])
            lx[j]-=d;
            for(int j=1;j<=ny;j++)
            {
                if(visy[j]) ly[j]+=d;
                else slack[j]-=d;
            }
        } 
    }   
    int res=0;
    for(int i=1;i<=ny;i++)
    if(cy[i]!=-1)
    res+=link[cy[i]][i];
    return res;
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            scanf("%d",&link[i][j]);
        nx=ny=n;
        int ans=KM();
        printf("%d\n",ans); 
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值