码迷,mamicode.com
首页 > 其他好文 > 详细

奔小康赚大钱---hdu2255(最大带权匹配)

时间:2015-08-10 13:32:10      阅读:103      评论:0      收藏:0      [点我收藏+]

标签:

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2255

带权匹配问题的模板;

运用KM算法;

技术分享
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#define INF 0xfffffff
#define N 330
using namespace std;

int maps[N][N], visx[N], visy[N], used[N], lx[N], ly[N], s[N], n;
///visx[i]代表第i人是否在曾广路上,
///used[i]代表第i个村庄是否被占用;
///lx[],ly[]代表人和村庄的顶标;
bool Find(int u)
{
    visx[u] = 1;
    for(int i=1; i<=n; i++)
    {
        if(!visy[i] && lx[u]+ly[i] == maps[u][i])
        {
            visy[i] = 1;
            if(!used[i] || Find(used[i]))
            {
                used[i]=u;
                return true;
            }
        }
        else
        {
            s[i] = min(s[i], lx[u]+ly[i] - maps[u][i]);
        }
    }
    return false;
}
int KM()
{
    memset(used, 0, sizeof(used));
    memset(lx, 0, sizeof(lx));
    memset(ly, 0, sizeof(ly));
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
            lx[i] = max(maps[i][j], lx[i]);///初始化人的顶标;
    }
    for(int i=1; i<=n; i++)///寻找最大匹配;人
    {
        for(int j=1; j<=n; j++)
            s[j]=INF;
        while(1)
        {
            memset(visx, 0, sizeof(visx));
            memset(visy, 0, sizeof(visy));
            if(Find(i))///找到的曾广路就退出,否则改变顶标,找到为止;
                break;
            int d = INF;
            for(int j=1; j<=n; j++)
            {
                if(!visy[j])
                    d=min(d, s[j]);
            }
            for(int j=1; j<=n; j++)
            {
                if(visx[j])
                    lx[j]-=d;
                if(visy[j])
                    ly[j]+=d;
            }
        }
    }
    int ans=0;
    for(int i=1; i<=n; i++)
    {
        ans+=maps[used[i]][i];
    }
    return ans;
}

int main()
{
    while(scanf("%d", &n)!=EOF)
    {
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
                scanf("%d", &maps[i][j]);
        }
        printf("%d\n", KM());
    }
    return 0;
}
View Code

 

奔小康赚大钱---hdu2255(最大带权匹配)

标签:

原文地址:http://www.cnblogs.com/zhengguiping--9876/p/4717717.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!