KM算法(二分图的最大权匹配)

结合例子来学模板
奔小康赚大钱 HDU - 2255

const int maxn=307;
const int INF=0x3f3f3f3f;
int love[maxn][maxn];   // 记录每个妹子和每个男生的好感度
int ex_girl[maxn];      // 每个妹子的期望值
int ex_boy[maxn];       // 每个男生的期望值
bool vis_girl[maxn];    // 记录每一轮匹配匹配过的女生
bool vis_boy[maxn];     // 记录每一轮匹配匹配过的男生
int match[maxn];        // 记录每个男生匹配到的妹子 如果没有则为-1
int slack[maxn];        // 记录每个汉子如果能被妹子倾心最少还需要多少期望值
int N;
bool dfs(int girl)
{
    vis_girl[girl]=true;
    for(int boy=0;boy<N;boy++){
        //每一轮匹配 每个男生只尝试一次
        if(vis_boy[boy])
            continue;
        int gap=ex_girl[girl]+ex_boy[boy]-love[girl][boy];
        //如果符合要求
        if(gap==0){
            vis_boy[boy]=true;
            // 找到一个没有匹配的男生 或者该男生的妹子可以找到其他人
            if(match[boy]==-1||dfs(match[boy])){
                match[boy]=girl;
                return true;
            }
        }
        //slack 可以理解为该男生要得到女生的倾心 还需多少期望值 取最小值备胎的样子
        else if(slack[boy]>gap)
            slack[boy]=gap;
    }
    return false;
}
int KM()
{
    //初始每个男生都没有匹配的女生
    memset(match,-1,sizeof(match));
    //初始每个男生的期望值为0
    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++)
            if(love[i][j]>ex_girl[i])
                ex_girl[i]=love[i][j];
    }
    //尝试为每一个女生解决归宿问题
    for(int i=0;i<N;i++){
        //因为要取最小值 初始化为无穷大
        for(int j=0;j<=N;j++)
            slack[j]=INF;
        // 为每个女生解决归宿问题的方法是 :如果找不到就降低期望值,直到找到为止
        while(true){
            //记录每轮匹配中男生女生是否被尝试匹配过
            memset(vis_girl,false,sizeof(vis_girl));
            memset(vis_boy,false,sizeof(vis_boy));
            //找到归宿
            if(dfs(i))
                break;
            //如果不能找到就降低期望值
            //最小可降低的期望值
            int d=INF;
            for(int j=0;j<N;j++)
                if(!vis_boy[j]&&d>slack[j])
                    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;
                //没有访问过的boy 因为girl们的期望值降低,距离得到女生倾心又进了一步!
                else
                    slack[j]-=d;
            }
        }
    }
    // 匹配完成求出所有配对的好感度的和
    int res=0;
    for(int i=0;i<N;i++)
        res+=love[match[i]][i];
    return res;
}
int main()
{
    while(~scanf("%d",&N)){
        for(int i=0;i<N;i++)
            for(int j=0;j<N;j++)
                scanf("%d",&love[i][j]);
        printf("%d\n", KM());
    }
    return 0;
}

拓展应用
Engaging with Loyal Customers Kattis - engaging

# define INF 0x3f3f3f3f
const int maxn=2e3+7;
int love[maxn][maxn];
int ex_girl[maxn],ex_boy[maxn];
bool vis_girl[maxn],vis_boy[maxn];
int match[maxn];
int slack[maxn];
int nx,ny;
vector<pair<int,int> >vec;
bool dfs(int girl)
{
    vis_girl[girl]=true;
    for(int boy=1;boy<=ny;boy++){
        if(vis_boy[boy])
            continue;
        int gap=ex_girl[girl]+ex_boy[boy]-love[girl][boy];
        if(gap==0){
            vis_boy[boy]=true;
            if(match[boy]==-1||dfs(match[boy])){
                match[boy]=girl;
                return true;
            }
        }
        else
            slack[boy]=min(slack[boy],gap);
    }
    return false;
}
pair<int,int> KM()
{
    memset(match,-1,sizeof(match));
    memset(ex_boy,0,sizeof(ex_boy));
    for(int i=1;i<=nx;i++){
        ex_girl[i]=-INF;
        for(int j=1;j<=ny;j++)
            ex_girl[i]=max(ex_girl[i],love[i][j]);
    }
    for(int i=1;i<=nx;i++){
        for(int j=1;j<=ny;j++)
            slack[j]=INF;
        while(true){
            memset(vis_girl,false,sizeof(vis_girl));
            memset(vis_boy,false,sizeof(vis_boy));
            if(dfs(i))
                break;
            int d=INF;
            for(int j=1;j<=ny;j++)
                if(!vis_boy[j])
                    d=min(d,slack[j]);
            for(int j=1;j<=nx;j++){
                if(vis_girl[j])
                    ex_girl[j]-=d;
            }
            for(int j=1;j<=ny;j++){
                if(vis_boy[j])
                    ex_boy[j]+=d;
                else
                    slack[j]-=d;
            }
        }
    }
    int sum=0,num=0;
    for(int i=1;i<=ny;i++){
        if(match[i]>0){
            sum+=love[match[i]][i];
            num++;
            vec.push_back(make_pair(match[i],i));
        }
    }
    return make_pair(sum,num);
}
int main()
{
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    bool flag_change=true;
    nx=n;
    ny=m;
    if(n>m){
        nx=m;
        ny=n;
        flag_change=false;
    }
    while(k--){
        int customers,gift,level;
        scanf("%d%d%d",&customers,&gift,&level);
        if(!flag_change)
            love[customers][gift]=level;
        else
            love[gift][customers]=level;
    }
    pair<int,int> temp=KM();
    printf("%d\n%d\n",temp.first,temp.second);
    int len=vec.size();
    for(int i=0;i<len;i++){
        if(flag_change)
            printf("%d %d\n",vec[i].second,vec[i].first);
        else
            printf("%d %d\n",vec[i].first,vec[i].second);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值