匈牙利算法的matlab实现

匈牙利算法

  • 算法简介
  • 算法原理
  • 算法实现(附代码)
  • 测试

算法简介

下面摘用百度百科中的解释。
匈牙利算法(Hungarian method)是由匈牙利数学家Edmonds于1965年提出,因而得名。匈牙利算法是基于Hall定理中充分性证明的思想,它是二分图匹配最常见的算法,该算法的核心就是寻找增广路径,它是一种用增广路径求二分图最大匹配的算法。

简单来说,匈牙利算法就是为了解决匹配问题的一种算法。
可以设想这样一个问题,有三个工人要共同完成三件任务。每个人只能专注完成一件任务。且工人完成不同任务所需要的时间是不一样的。当然这里可以直接通过枚举的方式来遍历所有的分配方式来进行求解。但当求解的问题的维度变得比较大时,这样处理就显得不太明智了。所以,为了解决这一类问题,匈牙利算法得以提出。

算法原理

算法的输入时一个代价矩阵 c c ,其中 cij c i j 表示工人i在任务j上的工作时长。

算法总的来说分成几步。

step1

将矩阵 c c 化为每行每列都至少有一个0的矩阵。
例如输入矩阵 c=(1243) c = ( 1 4 2 3 )
首先处理行
找到每行中的最小1元素,如第一行中找到的是1,第二行中是2。每行中所有元素都减去该最小元素得 (0031) ( 0 3 0 1 )
之后处理列
这里检查列发现第二列中没有0元素。故继续找到第二列总最小元素,并用第二列中的元素减去该最小元素,得 (0020) ( 0 2 0 0 ) .

step2

检查矩阵中是否有相互独立0的元素。也就是说能否找到与矩阵维度相同数目的0,它们占据不同的行列位置。如上的例子就可以找到这样的两个零元素, c11 c 11 c22 c 22 。当然也有不是这种情况的例子。如[

02090531086005032070620542 ( 0 5 0 2 2 2 3 0 0 0 0 10 5 7 5 9 8 0 0 4 0 6 3 6 2 )
]
如果在该步能找到独立零,则算法就可在此结束。如上的例子,可以输出 (1122) ( 1 2 1 2 ) .表示工人1完成任务1,工人2完成任务2。否则继续执行下面步骤。

step3

以每一个0元素为中心画十字架(即选中所有零元素所在的行元素和列元素)。找出没有被选中的元素。然后将没有被选中的元素所在的行进行标记。然后执行以下两个步骤

1. 对以标记的行中的零元素所在的列进行标记。
2. 对已标记的列中的零元素所在的行进行标记。

重复进行上述操作,直至不能再进行标记为止。
可以证明,没有标记的行以及标记了的列可以覆盖所有的零元素。

step4

对没有覆盖的元素,找到里面值最小的一个。将标记了的行中的所有元素减去这个值,将标记了的列中的元素加上这个值。这样的操作可以确保在未被覆盖的区域内至少产生一个0,且原本的0元素不会改变。
之后跳转到step2。

代码

传送门
简要说明一下。Hungarian.m文件是主程序。
两个辅助程序:
1. fix.m用于查找最佳分配,如果答案则可在all中得到所有最佳匹配。否则all为空,result中保存次最佳匹配。
2. findMin.m用于找到指定下标范围内的最小元素。

测试结果

输入
输入1
输出
输出1
输入
输入2
输出
这里写图片描述

你可能感兴趣的:(matlab)