hdoj 2255 奔小康赚大钱 【KM算法模板题】

奔小康赚大钱

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 5339    Accepted Submission(s): 2334

Problem Description

传说在遥远的地方有一个非常富裕的村落,有一天,村长决定进行制度改革：重新分配房子。
这可是一件大事,关系到人民的住房问题啊。村里共有n间房间,刚好有n家老百姓,考虑到每家都要有房住（如果有老百姓没房子住的话，容易引起不安定因素），每家必须分配到一间房子且只能得到一间房子。
另一方面,村长和另外的村领导希望得到最大的效益,这样村里的机构才会有钱.由于老百姓都比较富裕,他们都能对每一间房子在他们的经济范围内出一定的价格,比如有3间房子,一家老百姓可以对第一间出10万,对第2间出2万,对第3间出20万.(当然是在他们的经济范围内).现在这个问题就是村领导怎样分配房子才能使收入最大.(村民即使有钱购买一间房子但不一定能买到,要看村领导分配的).

 

Input

输入数据包含多组测试用例，每组数据的第一行输入n,表示房子的数量(也是老百姓家的数量)，接下来有n行,每行n个数表示第i个村名对第j间房出的价格(n<=300)。

 

Output

请对每组数据输出最大的收入值，每组的输出占一行。

 

Sample Input

  
  

   
   2
100 10
15 23
  
  

 

Sample Output

  
  

   
   123
  
  

 

KM算法入门题目，抽空补上KM模板。

AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN 400
#define INF 100000000//注意INF值要比所有边权值大
using namespace std;
int match[MAXN];//匹配
int lx[MAXN], ly[MAXN];//顶标
int slack[MAXN];//记录T集中节点的松弛量
int Map[MAXN][MAXN];//存储原图S-T节点间的关系
bool visx[MAXN], visy[MAXN];
int nx, ny;//S集中节点数目 T集中节点数目
int DFS(int x)
{
    visx[x] = true;
    for(int y = 0; y < ny; y++)
    {
        if(visy[y]) continue;
        int t = lx[x] + ly[y] - Map[x][y];//判断边<x,y>是否在相等子图里面
        if(t == 0)//边<x,y>在相等子图里面
        {
            visy[y] = true;
            if(match[y] == -1 || DFS(match[y]))
            {
                match[y] = x;
                return 1;
            }
        }
        else if(slack[y] > t)// 更新松弛量
            slack[y] = t;
    }
    return 0;
}
int KM()
{
    memset(match, -1, sizeof(match));
    memset(ly, 0, sizeof(ly));
    for(int x = 0; x < nx; x++)//lx[x]初始化为Map[x][y]中最大值
    {
        lx[x] = -INF;
        for(int y = 0; y < ny; y++)
        {
            if(Map[x][y] > lx[x])
                lx[x] = Map[x][y];
        }
    }
    for(int x = 0; x < nx; x++)//匹配
    {
        for(int i = 0; i < ny; i++)//初始化T集中所有节点的 松弛量
            slack[i] = INF;
        while(1)
        {
            memset(visx, false, sizeof(visx));
            memset(visy, false, sizeof(visy));
            if(DFS(x)) break;//匹配成功
            //匹配失败
            int d = INF;
            //寻找最小的松弛量d
            for(int i = 0; i < ny; i++)
            {
                if(!visy[i] && d > slack[i])
                    d = slack[i];
            }
            //修改顶标 为了有更大机会完美匹配
            //S集里面所有点 全部减去最小松弛量d
            for(int i = 0; i < nx; i++)
            {
                if(visx[i])
                    lx[i] -= d;
            }
            //为了保证S-T的匹配不离开相等子图即对边<x,y>依旧有lx[x] + ly[y] == Map[x][y]
            //T集里面所有点 全部增加最小松弛量d
            for(int i = 0; i < ny; i++)
            {
                if(visy[i]
                    ly[i] += d;
                else//节点的松弛量 减少d
                    slack[i] -= d;
            }
        }
    }
    int res = 0;
    for(int i = 0; i < ny; i++)
    {
        if(match[i] != -1)//有匹配
            res += Map[match[i]][i];//累加权值
    }
    return res;
}
int main()
{
    int N;
    while(scanf("%d", &N) != EOF)
    {
        nx = ny = N;
        for(int i = 0; i < N; i++)
            for(int j = 0; j < N; j++)
                scanf("%d", &Map[i][j]);
        int ans = KM();
        printf("%d\n", ans);
    }
    return 0;
}