二分图匹配模板

最大匹配  邻接矩阵

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int N=1500;//N不能太大 否则超时
int cap[N][N];//初始化要清零
int _link[N];
bool used[N];
int nx,ny;//1->nx
bool _find(int t)
{
    for(int i=1;i<=ny;i++)

    if(!used[i]&&cap[t][i]==1)
    {
        used[i]=true;
        if(_link[i]==-1||_find(_link[i]))
        {
            _link[i]=t;
            return true;
        }
    }
    return false;
}
int MaxMatch()
{
    int num=0;
    memset(_link,-1,sizeof(_link));
    for(int i=1;i<=nx;i++)
    {
        memset(used,false,sizeof(used));
        if(_find(i))   num++;
    }
    return num;
}

 

最大匹配 邻接表

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=10010;
/*初始化

注意 加边时候X和Y统一标号,不是分别标号

加边时按照总点数来加
cnt=0;
memset(head,-1,sizeof(head));
*/
int n;//X点数
int link[N];
bool used[N];
int head[N],cnt;
struct TT
{
    int v,next;
}edge[50000];
void addedge(int u,int v)
{
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
bool can(int t)
{
    for(int p=head[t];p!=-1;p=edge[p].next)
    {
        int v=edge[p].v;
        if(used[v]==false)
        {
            used[v]=1;
            if(link[v]==-1||can(link[v]))
            {
                link[v]=t;
                return true;
            }
        }
    }
    return false;
}
int MaxMatch()
{
    int num=0;
    memset(link,-1,sizeof(link));
    for(int i=1;i<=n;i++)
    {
        memset(used,false,sizeof(used));
        if(can(i)) num++;
    }
    return num;
}

 

多重匹配 邻接矩阵

#include<cstdio>
#include<cstring>
int n,m;//X和Y
bool mat[1001][501],used[501];//mat初始清零
int link[1001][501];
int mid;
bool can(int t)
{
    for(int i=1;i<=m;i++)
    {
        if(!used[i]&&mat[t][i])
        {
            used[i]=1;
            if(link[i][0]<mid)
            {
                link[i][++link[i][0]]=t;
                return true;
            }
            else
            {
                for(int j=1;j<=link[i][0];j++)
                if(can(link[i][j]))
                {
                    link[i][j]=t;
                    return true;
                }
            }
        }
    }
    return false;
}
bool Match()//X中所有点都匹配
{
    for(int i=0;i<=m;i++)  link[i][0]=0;
    for(int i=1;i<=n;i++)
    {
        memset(used,false,sizeof(used));
        if(!can(i))  return false;
    }
    return true;
}

最优完备匹配 KM算法

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
const int N=210;
const int inf=(1<<28);
//w是完全图,若i->j没有路,则w[i][j]=inf;
//若有多条路,则取最小路
int n,m;//点数1->n,边数
int match[N],lack,w[N][N],lx[N],ly[N];;
bool visx[N],visy[N];
bool dfs(int x){
    visx[x]=true;
    for(int y=1; y<=n; y++){
        if(visy[y]) continue;
        int t=lx[x]+ly[y]-w[x][y];
        if(t==0){
            visy[y]=true;
            if(match[y]==-1||dfs(match[y])){
                match[y]=x;
                return true;
            }
        }
        else if(lack>t) lack=t;
    }
    return false;
}
int KM(){
 int i,j;
 for(i=0;i<=n;i++)lx[i]=0;
 for(i=0;i<=n;i++)ly[i]=0;
    memset(match,-1,sizeof(match));
    for(i=1; i<=n; i++)for( j=1; j<=n; j++)if(w[i][j]>lx[i]) lx[i]=w[i][j];
    for(int x=1; x<=n; x++){
        while(1){
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            lack=inf;
            if(dfs(x)) break;
            for(i=1; i<=n; i++){
                if(visx[i]) lx[i]-=lack;
                if(visy[i]) ly[i]+=lack;
            }
        }
    }
    int total=0;
    for(i=1;i<=n;i++)
    {
        total+=w[match[i]][i];
    }
    return total;
}

KM算法小优化

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
const int N=1004;
const int inf=(1<<28);
int n,m;//点数1->n,边数
int match[N],lack,w[N][N],lx[N],ly[N];;
bool visx[N],visy[N];
bool dfs(int x){
    visx[x]=true;
    for(int y=1; y<=n; y++){
        if(visy[y]) continue;
        int t=lx[x]+ly[y]-w[x][y];
        if(t==0){
            visy[y]=true;
            if(match[y]==-1||dfs(match[y])){
                match[y]=x;
                return true;
            }
        }
    }
    return false;
}
int KM(){
 int i,j;
 memset(ly,0,sizeof(ly));
    memset(match,-1,sizeof(match));
    for(i=1; i<=n; i++)
    {
        lx[i]=-inf;
        for( j=1; j<=n; j++)if(w[i][j]>lx[i]) lx[i]=w[i][j];
    }
    for(int x=1; x<=n; x++){
        while(1){
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            lack=inf;
            if(dfs(x)) break;
            for(i=1; i<=n; i++)
                if(visx[i])
                    for(j=1; j<=n; j++)
                        if(!visy[j])
                            lack=min(lack,lx[i]+ly[j]-w[i][j]);
            for(i=1; i<=n; i++){
                if(visx[i]) lx[i]-=lack;
                if(visy[i]) ly[i]+=lack;
            }
        }
    }
    int total=0;
    for(i=1;i<=n;i++)
    {
        total+=w[match[i]][i];
    }
    return total;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值